Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def load_trained_model():
|
|
24 |
filename="best_model_compressed.pt",
|
25 |
token=os.getenv('HF_TOKEN')
|
26 |
)
|
27 |
-
checkpoint = torch.load(model_path, map_location=config.device
|
28 |
model.load_state_dict(checkpoint['model_state_dict'])
|
29 |
model.to(config.device)
|
30 |
model.eval()
|
|
|
24 |
filename="best_model_compressed.pt",
|
25 |
token=os.getenv('HF_TOKEN')
|
26 |
)
|
27 |
+
checkpoint = torch.load(model_path, map_location=config.device)
|
28 |
model.load_state_dict(checkpoint['model_state_dict'])
|
29 |
model.to(config.device)
|
30 |
model.eval()
|