storesource commited on
Commit
fd1e932
1 Parent(s): dd0f7c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- urllib.request.urlretrieve(url, temp_file.name)
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