Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ def load_model():
|
|
| 28 |
|
| 29 |
# Load weights - using model_state.bin instead of pytorch_model.bin
|
| 30 |
state_dict = torch.hub.load_state_dict_from_url(
|
| 31 |
-
"https://huggingface.co/KenLumod/ML-
|
| 32 |
map_location=device
|
| 33 |
)
|
| 34 |
model.load_state_dict(state_dict)
|
|
|
|
| 28 |
|
| 29 |
# Load weights - using model_state.bin instead of pytorch_model.bin
|
| 30 |
state_dict = torch.hub.load_state_dict_from_url(
|
| 31 |
+
"https://huggingface.co/KenLumod/ML-Fake-Real-News-Detector-Final/resolve/main/model_state.bin",
|
| 32 |
map_location=device
|
| 33 |
)
|
| 34 |
model.load_state_dict(state_dict)
|