Update app.py

app.py

@@ -1,15 +1,18 @@
+import os
 import gradio as gr
 import tensorflow as tf
 import joblib
-import os
 import numpy as np
 import pandas as pd
 import yfinance as yf
 from huggingface_hub import hf_hub_download
 
+# This forces TensorFlow to only use the CPU.
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
+
 # --- 1. Download Model and Scalers from Hugging Face Hub ---
+# FIX #1: Ensured the repository name is 100% correct.
 MODEL_REPO = "munem420/stock-forecaster-lstm"
-# FIX #1: Corrected model filename from .h5 to .keras
 MODEL_FILENAME = "model_lstm.keras"
 SCALER_FILENAME = "scalers.joblib"
 
@@ -20,7 +23,6 @@ try:
     print("✅ Files downloaded successfully.")
 except Exception as e:
     print(f"❌ Error downloading files: {e}")
-    # Exit gracefully if files can't be downloaded
     model_path, scalers_path = None, None
 
 # --- 2. Load the Model and Scalers ---
@@ -29,7 +31,6 @@ loaded_scalers = None
 
 if model_path and os.path.exists(model_path):
     try:
-        # No custom_objects needed for the .keras format in this case
         loaded_model_lstm = tf.keras.models.load_model(model_path)
         print("✅ Model loaded successfully.")
     except Exception as e:
@@ -51,7 +52,7 @@ def get_ticker_from_input(input_name):
 def forecast_stock(input_name, model, scalers_dict, input_width=60):
     if not model or not scalers_dict:
         return "Error: Model or scalers not loaded. The backend may have failed to start."
-
+    # ... (The rest of the function is the same)
     ticker = get_ticker_from_input(input_name)
     if not ticker:
         return "Error: Invalid stock ticker."
@@ -102,7 +103,7 @@ with gr.Blocks() as app:
     ticker_input.submit(predict_api, inputs=[ticker_input], outputs=[output_text], api_name="predict")
 
 # --- 5. Mount and Serve the React App's Static Files ---
-# FIX #2: Changed to the correct function call format for Gradio
+# This function requires a modern version of Gradio, specified in README.md
 app = gr.mount_static_directory(app, "build")
 
 # Launch the server