Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -201,7 +201,7 @@ def load_model_and_vocabs():
|
|
| 201 |
dropout=0.3,
|
| 202 |
upsample_factor=2
|
| 203 |
).to(device)
|
| 204 |
-
|
| 205 |
# Load trained weights
|
| 206 |
model.load_state_dict(torch.load('best_model.pth', map_location=device))
|
| 207 |
model.eval()
|
|
|
|
| 201 |
dropout=0.3,
|
| 202 |
upsample_factor=2
|
| 203 |
).to(device)
|
| 204 |
+
|
| 205 |
# Load trained weights
|
| 206 |
model.load_state_dict(torch.load('best_model.pth', map_location=device))
|
| 207 |
model.eval()
|