fschwartzer commited on
Commit
c65f0dc
·
verified ·
1 Parent(s): b77796f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -7,6 +7,10 @@ df = pd.read_csv('anomalies.csv', sep=',', decimal='.')
7
 
8
  # Function to generate a response
9
  def response(question):
 
 
 
 
10
  tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
11
  final_rresposta = tqa(table=df, query=question)['cells'][0]
12
  return final_response
 
7
 
8
  # Function to generate a response
9
  def response(question):
10
+ question = user_question # ou o que você está passando como input
11
+ if not isinstance(question, str):
12
+ raise TypeError(f"Esperado uma string para a pergunta, mas recebeu {type(question)}")
13
+
14
  tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
15
  final_rresposta = tqa(table=df, query=question)['cells'][0]
16
  return final_response