Update app.py

app.py

@@ -9,7 +9,7 @@ st.title("Stock Trend Predictor: Bullish or Bearish?")
 # Load pre-trained model from Hugging Face
 @st.cache_resource
 def load_model():
-    model_name = "Kaylah072001/stock_prediction_model"  # Replace with your actual model name on Hugging Face
+    model_name = "Kaylah072001/stock_prediction_model.h5"  # Replace with your actual model name on Hugging Face
     try:
         model = AutoModelForImageClassification.from_pretrained(model_name)
         feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)