CCockrum commited on
Commit
922c0b2
·
verified ·
1 Parent(s): 6dd319e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -156,8 +156,10 @@ def generate_summary(info, ratios):
156
  return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
157
 
158
  def answer_investing_question(question):
159
- response = chat_model(['generated_text'])
160
- return response
 
 
161
 
162
 
163
  def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
 
156
  return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
157
 
158
  def answer_investing_question(question):
159
+ response = chat_model(question)[0]['generated_text']
160
+ cleaned_response = response.replace("Answer:", "").strip()
161
+ return cleaned_response
162
+
163
 
164
 
165
  def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):