hamoudaGue commited on
Commit
69d4d1e
·
verified ·
1 Parent(s): 11daa20

Update app/app.py

Browse files
Files changed (1) hide show
  1. app/app.py +1 -1
app/app.py CHANGED
@@ -50,7 +50,7 @@ def load_model(name: str):
50
 
51
  # Load model - HF Spaces will have these files in the repo
52
  try:
53
- model_path = os.path.join("models", model_filename)
54
  mdl.load_state_dict(torch.load(model_path, map_location=device, weights_only=True))
55
  except Exception as e:
56
  st.error(f"❌ Error loading model: {e}")
 
50
 
51
  # Load model - HF Spaces will have these files in the repo
52
  try:
53
+ model_path = os.path.join(".", model_filename)
54
  mdl.load_state_dict(torch.load(model_path, map_location=device, weights_only=True))
55
  except Exception as e:
56
  st.error(f"❌ Error loading model: {e}")