40bqa / app.py
wiwide's picture
Update app.py
c633bb4
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()