Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ INPUT_SHAPE = (256, 256, 3)
|
|
| 23 |
|
| 24 |
# Load model from Hugging Face Hub
|
| 25 |
try:
|
| 26 |
-
model_path = hf_hub_download(repo_id="drprs-
|
| 27 |
model = keras.models.load_model(model_path)
|
| 28 |
except Exception as e:
|
| 29 |
raise RuntimeError(f"Error loading model: {str(e)}")
|
|
|
|
| 23 |
|
| 24 |
# Load model from Hugging Face Hub
|
| 25 |
try:
|
| 26 |
+
model_path = hf_hub_download(repo_id="drprs-research/petra", filename="petra.keras")
|
| 27 |
model = keras.models.load_model(model_path)
|
| 28 |
except Exception as e:
|
| 29 |
raise RuntimeError(f"Error loading model: {str(e)}")
|