Update app.py

app.py

@@ -16,6 +16,18 @@ labels = {
     'amphibia': ['American Bullfrog', 'American Toad', 'Green Frog', 'Northern Leopard Frog']
 }
 
+model_url = "https://github.com/sowbaranika1302/ohio_classifier/raw/main/Models/inceptionv3_best_withoutblanks.h5"
+model_path = "inceptionv3_best_withoutblanks.h5"
+
+# Download the model if it doesn't exist locally
+if not os.path.exists(model_path):
+    response = requests.get(model_url)
+    with open(model_path, 'wb') as f:
+        f.write(response.content)
+
+# Load the model
+model = load_model(model_path)
+
 # Preprocess the image
 def preprocess_image(image):
     img = image.resize((224, 224))  # MobileNet requires 224x224 input size
@@ -27,7 +39,7 @@ def preprocess_image(image):
 # Function to perform inference
 def predict(img):
     img_array, resized_img = preprocess_image(img)
-    model = load_model(r"inceptionv3_classs.h5")
+    #model = load_model(r"inceptionv3_classs.h5")
     preds = model.predict(img_array)  # Get the model predictions
     decoded_preds = np.argmax(preds)