Update app.py

app.py

@@ -9,7 +9,8 @@ from transformers import pipeline
 
 # Initialize Models
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-chat_model = pipeline("text-generation", model="google/flan-t5-large", max_length=256)
+chat_model = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_length=256)
+
 
 # API Key
 POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")
@@ -157,8 +158,8 @@ def generate_summary(info, ratios):
 
 def answer_investing_question(question):
     response = chat_model(question)[0]['generated_text']
-    cleaned_response = response.replace("Answer:", "").strip()
-    return cleaned_response
+    return response
+