|
import requests |
|
from bs4 import BeautifulSoup |
|
from transformers import pipeline |
|
import gradio as gr |
|
|
|
def get_web_page(url): |
|
response = requests.get(url) |
|
soup = BeautifulSoup(response.text, 'html.parser') |
|
return soup.text |
|
|
|
def answer_question(question, context): |
|
qa_pipeline = pipeline("question-answering", model="tiiuae/falcon-40b") |
|
result = qa_pipeline({ |
|
'context': context, |
|
'question': question |
|
}) |
|
return result['answer'] |
|
|
|
def app(symbol, question): |
|
url = f"https://finance.yahoo.com/quote/{symbol}" |
|
context = get_web_page(url) |
|
answer = answer_question(question, context) |
|
return answer |
|
|
|
iface = gr.Interface(fn=app, inputs=["text", "text"], outputs="text") |
|
iface.launch() |
|
|