Update app.py

app.py

@@ -18,7 +18,7 @@ def load_model():
     with tempfile.NamedTemporaryFile(suffix=".keras") as temp_file:
         url = "https://huggingface.co/tomeheya/IM-417-128x128-classification/raw/main/IM_417_128.keras"
         urlretrieve(url, temp_file.name)
-        return tf.keras.models.load_model(temp_file.name) 
+        return keras.models.load_model(temp_file.name) 
     
 
 # Function to preprocess the input image