Update app/app.py
Browse files- 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("
|
| 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}")
|