Update app.py

app.py

@@ -12,10 +12,20 @@ data = {
 table = pd.DataFrame.from_dict(data)
 
 # tapex accepts uncased input since it is pre-trained on the uncased corpus
-query = "how many different countries had election in 21st century?"
-encoding = tokenizer(table=table, query=query, return_tensors="pt")
+#query = "how many different countries had election in 21st century?"
+#encoding = tokenizer(table=table, query=query, return_tensors="pt")
 
-outputs = model.generate(**encoding)
+# outputs = model.generate(**encoding)
 
-print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
-# [' 2008.0']
+# print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
+# [' 2008.0']
+
+def launch(input):
+    encoding = tokenizer(table=table, query=input, return_tensors="pt")
+    out=model.generate(**encoding)
+    return tokenizer.batch_decode(outputs, skip_special_tokens=True)
+
+iface = gr.Interface(launch,
+                    inputs="text",
+                    outputs="text")
+iface.launch(share=True)