CCockrum commited on
Commit
a6edca8
·
verified ·
1 Parent(s): e314039

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -8,10 +8,8 @@ import tempfile
8
  from transformers import pipeline
9
 
10
 
11
- # Lightweight model for quick answers
12
  light_chat_model = pipeline("text-generation", model="google/flan-t5-small", max_length=256)
13
-
14
- # Heavier model for advanced summarization
15
  advanced_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
16
 
17
  def answer_investing_question(question, level="basic"):
@@ -22,6 +20,7 @@ def answer_investing_question(question, level="basic"):
22
  response = advanced_summarizer(prompt, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
23
  return response.strip()
24
 
 
25
  # API Key
26
  POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")
27
 
@@ -156,9 +155,6 @@ def generate_summary(info, ratios):
156
  )
157
  return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
158
 
159
- def answer_investing_question(question):
160
- response = chat_model(question)[0]['generated_text']
161
- return response
162
 
163
  def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
164
  info = get_company_info(symbol)
 
8
  from transformers import pipeline
9
 
10
 
11
+ # Load Two Models
12
  light_chat_model = pipeline("text-generation", model="google/flan-t5-small", max_length=256)
 
 
13
  advanced_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
14
 
15
  def answer_investing_question(question, level="basic"):
 
20
  response = advanced_summarizer(prompt, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
21
  return response.strip()
22
 
23
+
24
  # API Key
25
  POLYGON_API_KEY = os.getenv("POLYGON_API_KEY")
26
 
 
155
  )
156
  return summarizer(report, max_length=250, min_length=100, do_sample=False)[0]['summary_text']
157
 
 
 
 
158
 
159
  def stock_research(symbol, eps=5.0, growth=0.1, book=500000000):
160
  info = get_company_info(symbol)