Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -31,6 +31,7 @@ model = BartForConditionalGeneration.from_pretrained("microsoft/tapex-large-fine
|
|
31 |
|
32 |
#new
|
33 |
pipe_tapas = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
|
|
|
34 |
|
35 |
|
36 |
|
@@ -45,18 +46,21 @@ def process2(query, csv_dataStr):
|
|
45 |
result_tapex=tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
46 |
#google
|
47 |
result_tapas = pipe_tapas(table=table, query=query)['cells'][0]
|
48 |
-
|
|
|
|
|
49 |
|
50 |
|
51 |
# Inputs
|
52 |
-
query_text = gr.Text(label="
|
53 |
# input_file = gr.File(label="Upload a CSV file", type="file")
|
54 |
-
input_data = gr.Text(label="
|
55 |
# rows_slider = gr.Slider(label="Number of rows")
|
56 |
|
57 |
# Output
|
58 |
answer_text_tapex = gr.Text(label="TAPEX answer")
|
59 |
answer_text_tapas = gr.Text(label="TAPAS answer")
|
|
|
60 |
|
61 |
description = "This Space lets you ask questions on CSV documents with Microsoft [TAPEX-Large](https://huggingface.co/microsoft/tapex-large-finetuned-wtq) and Google [TAPAS-Large](https://huggingface.co/google/tapas-large-finetuned-wtq). \
|
62 |
Both have been fine-tuned on the [WikiTableQuestions](https://huggingface.co/datasets/wikitablequestions) dataset. \n\n\
|
@@ -74,7 +78,7 @@ iface = gr.Interface(
|
|
74 |
layout="vertical",
|
75 |
fn=process2,
|
76 |
inputs=[query_text, input_data],
|
77 |
-
outputs=[answer_text_tapex, answer_text_tapas],
|
78 |
examples=[
|
79 |
|
80 |
],
|
|
|
31 |
|
32 |
#new
|
33 |
pipe_tapas = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")
|
34 |
+
pipe_tapas2 = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wikisql-supervised")
|
35 |
|
36 |
|
37 |
|
|
|
46 |
result_tapex=tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
|
47 |
#google
|
48 |
result_tapas = pipe_tapas(table=table, query=query)['cells'][0]
|
49 |
+
#google2
|
50 |
+
result_tapas2 = pipe_tapas2(table=table, query=query)['cells'][0]
|
51 |
+
return result_tapex, result_tapas, result_tapas2
|
52 |
|
53 |
|
54 |
# Inputs
|
55 |
+
query_text = gr.Text(label="")
|
56 |
# input_file = gr.File(label="Upload a CSV file", type="file")
|
57 |
+
input_data = gr.Text(label="")
|
58 |
# rows_slider = gr.Slider(label="Number of rows")
|
59 |
|
60 |
# Output
|
61 |
answer_text_tapex = gr.Text(label="TAPEX answer")
|
62 |
answer_text_tapas = gr.Text(label="TAPAS answer")
|
63 |
+
answer_text_tapas2 = gr.Text(label="TAPAS wikisql answer")
|
64 |
|
65 |
description = "This Space lets you ask questions on CSV documents with Microsoft [TAPEX-Large](https://huggingface.co/microsoft/tapex-large-finetuned-wtq) and Google [TAPAS-Large](https://huggingface.co/google/tapas-large-finetuned-wtq). \
|
66 |
Both have been fine-tuned on the [WikiTableQuestions](https://huggingface.co/datasets/wikitablequestions) dataset. \n\n\
|
|
|
78 |
layout="vertical",
|
79 |
fn=process2,
|
80 |
inputs=[query_text, input_data],
|
81 |
+
outputs=[answer_text_tapex, answer_text_tapas, answer_text_tapas2],
|
82 |
examples=[
|
83 |
|
84 |
],
|