Spaces:

Haitam03
/

arabic-pronunciation-checker

Sleeping

Haitam03 commited on 25 days ago

Commit

a50c51e

verified ·

1 Parent(s): cfe88c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -266,7 +266,7 @@ def load_model():
         # Load weights
         model_weights = torch.hub.load_state_dict_from_url(
-            "https://huggingface.co/Haitam03/Wav2Vec2-mispronunciation-detector/resolve/main/epoch_3_full_data.pt",
             map_location=device
         )
         model.load_state_dict(model_weights)

         # Load weights
         model_weights = torch.hub.load_state_dict_from_url(
+            "https://huggingface.co/Haitam03/Wav2Vec2-mispronunciation-detector/resolve/main/best_mode_full_data.pt",
             map_location=device
         )
         model.load_state_dict(model_weights)