fschwartzer commited on
Commit
f736f9d
·
verified ·
1 Parent(s): 6e96ce7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,9 +7,8 @@ df = pd.read_excel('CG.xlsx')
7
 
8
  def response(user_question):
9
  tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
10
- resposta = tqa(table=df, query=user_question)['cells'][0]
11
- print(resposta)
12
- return str(resposta)
13
 
14
  # Streamlit interface
15
  st.markdown("""
 
7
 
8
  def response(user_question):
9
  tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
10
+ resposta = tqa(table=df, query=user_question)
11
+ return resposta
 
12
 
13
  # Streamlit interface
14
  st.markdown("""