Update app.py

app.py

@@ -472,7 +472,10 @@ import streamlit as st
 @st.cache_resource
 def load_soil_model():
     try:
-        model = torch.load("soil_best_model.pth", map_location=torch.device("cpu"))
+        # Define your SoilNet model architecture again
+        model = SoilNet()  # same architecture you trained with
+        state_dict = torch.load("soil_best_model.pth", map_location="cpu")
+        model.load_state_dict(state_dict)
         model.eval()
         return model
     except Exception as e: