Update app.py

app.py

@@ -9,15 +9,17 @@ from transformers import pipeline
 
 
 # Load Two Models
-light_chat_model = pipeline("text-generation", model="google/flan-t5-small", max_length=256)
+light_chat_model = pipeline(
+    "text-generation",
+    model="mistralai/Mistral-7B-Instruct-v0.2",
+    revision="refs/pr/23",  # Use latest stable revision if needed
+    use_auth_token="your_huggingface_access_token",
+    max_length=256
+)
 advanced_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-def answer_investing_question(question, level="basic"):
-    if level == "basic":
-        response = light_chat_model(question)[0]['generated_text']
-    else:
-        prompt = f"Explain in detail: {question}"
-        response = advanced_summarizer(prompt, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
+def answer_investing_question(question):
+    response = light_chat_model(question)[0]['generated_text']
     return response.strip()
 
 
@@ -200,12 +202,12 @@ with gr.Blocks(theme="soft") as iface:
             output_sector = gr.Dataframe()
         with gr.Tab("Historical Price Chart"):
             output_chart = gr.Plot()
-        with gr.Tab("Ask About Investing"):
-            user_question = gr.Textbox(label="Ask about investing...")
-            level = gr.Dropdown(choices=["basic", "advanced"], value="basic", label="Answer Detail Level")
-            answer_box = gr.Textbox(label="Answer")
-            ask_button = gr.Button("Get Answer")
-            ask_button.click(fn=answer_investing_question, inputs=[user_question, level], outputs=[answer_box])
+        with gr.Blocks(theme="soft") as iface:
+            with gr.Tab("Ask About Investing"):
+                user_question = gr.Textbox(label="Ask about investing...")
+                answer_box = gr.Textbox(label="Answer")
+                ask_button = gr.Button("Get Answer")
+                ask_button.click(fn=answer_investing_question, inputs=[user_question], outputs=[answer_box])
 
 
     submit_btn = gr.Button("Run Analysis")