Update app.py

app.py

@@ -31,7 +31,7 @@ btn = st.button("Predict")
 # Create a function to load model
 @st.cache(allow_output_mutation=True)
 def load_model():
-    model_path = './model_1_updated'  # Update with the correct file name
+    model_path = './model_1_updated.pth'  # Update with the correct file name
     
     # Load the model configuration
     config = AutoConfig.from_pretrained(model_path)