Update streamlit_app.py

streamlit_app.py

@@ -25,20 +25,13 @@ except:
 
 @st.cache_resource
 def load_model():
-    import os
-    import urllib.request
-
-    MODEL_URL = "https://huggingface.co/spaces/leynessa/libkamaja_id/resolve/main/butterfly_classifier.pth"
-
-
-
     MODEL_PATH = "butterfly_classifier.pth"
-
-    # Download if not already present
+    
+    # Check if model exists locally
     if not os.path.exists(MODEL_PATH):
-        st.info("Downloading model...")
-        urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
-
+        st.error("Model file not found. Please upload butterfly_classifier.pth to your space.")
+        return None
+    
     # Load the model
     model = models.resnet18(pretrained=False)
     model.fc = torch.nn.Linear(model.fc.in_features, len(class_names))