Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ class GenreCNN(nn.Module):
|
|
| 61 |
# ββ Load model once at startup ββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 62 |
model = GenreCNN().to(DEVICE)
|
| 63 |
model.load_state_dict(
|
| 64 |
-
torch.load("best_model
|
| 65 |
)
|
| 66 |
model.eval()
|
| 67 |
print("β Model loaded successfully")
|
|
|
|
| 61 |
# ββ Load model once at startup ββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 62 |
model = GenreCNN().to(DEVICE)
|
| 63 |
model.load_state_dict(
|
| 64 |
+
torch.load("best_model.pth", map_location=DEVICE)
|
| 65 |
)
|
| 66 |
model.eval()
|
| 67 |
print("β Model loaded successfully")
|