Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,10 +8,8 @@ import tempfile
|
|
8 |
from transformers import pipeline
|
9 |
|
10 |
|
11 |
-
#
|
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)
|