Update

app.py

@@ -22,32 +22,40 @@ pipe_tapas = pipeline(
 )
 
 
-def process(query, file, correct_answer, rows=20):
+def process(query, file, rows=20):
     table = pd.read_csv(file.name, header=0).astype(str)
     table = table[:rows]
     result_tapex = pipe_tapex(table=table, query=query)
     result_tapas = pipe_tapas(table=table, query=query)
-    return result_tapex["answer"], result_tapas["answer"], correct_answer
+    return result_tapex["answer"], result_tapas["answer"]
 
 
 # Inputs
 query_text = gr.Text(label="Enter a question")
-input_file = gr.File(label="Upload a CSV file", type="file")
+input_file = gr.File(label="Upload a CSV file", type="filepath")
 rows_slider = gr.Slider(label="Number of rows")
 
 # Output
 answer_text_tapex = gr.Text(label="TAPEX answer")
 answer_text_tapas = gr.Text(label="TAPAS answer")
 
-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). \
-Both have been fine-tuned on the [WikiTableQuestions](https://huggingface.co/datasets/wikitablequestions) dataset. \n\n\
-A sample file with football statistics is available in the repository: \n\n\
-* Which team has the most wins? Answer: Manchester City FC\n\
-* Which team has the most wins: Chelsea, Liverpool or Everton? Answer: Liverpool\n\
-* Which teams have scored less than 40 goals? Answer: Cardiff City FC, Fulham FC, Brighton & Hove Albion FC, Huddersfield Town FC\n\
-* What is the average number of wins? Answer: 16 (rounded)\n\n\
-You can also upload your own CSV file. Please note that maximum sequence length for both models is 1024 tokens, \
-so you may need to limit the number of rows in your CSV file. Chunking is not implemented yet."
+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). "
+    "Both have been fine-tuned on the "
+    "[WikiTableQuestions](https://huggingface.co/datasets/wikitablequestions) dataset. \n\n"
+    "A sample file with football statistics is available in the repository: \n\n"
+    "* Which team has the most wins? Answer: Manchester City FC\n"
+    "* Which team has the most wins: Chelsea, Liverpool or Everton? Answer: "
+    "Liverpool\n"
+    "* Which teams have scored less than 40 goals? Answer: Cardiff City FC, "
+    "Fulham FC, Brighton & Hove Albion FC, Huddersfield Town FC\n"
+    "* What is the average number of wins? Answer: 16 (rounded)\n\n"
+    "You can also upload your own CSV file. Please note that maximum sequence "
+    "length for both models is 1024 tokens, so you may need to limit the number "
+    "of rows in your CSV file. Chunking is not implemented yet."
+)
 
 iface = gr.Interface(
     theme="huggingface",
@@ -63,7 +71,11 @@ iface = gr.Interface(
             "default_file.csv",
             20,
         ],
-        ["Which teams have scored less than 40 goals?", "default_file.csv", 20],
+        [
+            "Which teams have scored less than 40 goals?", 
+            "default_file.csv", 
+            20
+        ],
         ["What is the average number of wins?", "default_file.csv", 20],
     ],
     allow_flagging="never",

requirements.txt

@@ -1,2 +1,4 @@
 transformers
 torch
+gradio
+pandas