Spaces:
Sleeping
Sleeping
Commit
•
fd1e932
1
Parent(s):
dd0f7c7
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ from urllib.request import urlretrieve
|
|
17 |
def load_model():
|
18 |
with tempfile.NamedTemporaryFile(suffix=".keras") as temp_file:
|
19 |
url = "https://huggingface.co/tomeheya/IM-417-128x128-classification/raw/main/IM_417_128.keras"
|
20 |
-
|
21 |
return tf.keras.models.load_model(temp_file.name)
|
22 |
|
23 |
|
|
|
17 |
def load_model():
|
18 |
with tempfile.NamedTemporaryFile(suffix=".keras") as temp_file:
|
19 |
url = "https://huggingface.co/tomeheya/IM-417-128x128-classification/raw/main/IM_417_128.keras"
|
20 |
+
urlretrieve(url, temp_file.name)
|
21 |
return tf.keras.models.load_model(temp_file.name)
|
22 |
|
23 |
|