Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
@@ -40,15 +39,15 @@ name_to_ticker = {
 
 def predict(stock_name):
     ticker = name_to_ticker[stock_name]
-    print(f"TICKER VALUE: {ticker}")
+
     # Load the model
     project = hopsworks.login(
-        api_key_value=os.environ['Hopsworks_API_Key']
+        api_key_value="<HOPSWORKS_API_KEY>",
+        #api_key_value=os.environ['Hopsworks_API_Key'] # For running on Huggingface spaces
     )
     mr = project.get_model_registry()
     model = mr.get_model("FinanceModel", version=11)
     saved_model_dir = model.download()
-    print(saved_model_dir)
     model = load_model(saved_model_dir + "/model.keras")
 
     # Fetch the data used to train the model
@@ -110,11 +109,11 @@ interface = gr.Interface(
         choices=list(name_to_ticker.keys()),
         value="ABB.ST",  # Default value
     ),
-    outputs=gr.Plot(label="Index Prediction Plot"),
-    title="Financial Index Predictor",
-    description="Enter the name of a financial index to generate a plot showing true values for the past 30 days and the predicted value.",
+    outputs=gr.Plot(label="Stock Prediction Plot"),
+    title="Stock Predictor",
+    description="Enter the name of a stock to generate a plot showing true values for the past 60 days and the predicted values for the next 5 days.",
 )
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()