update app.py

app.py

@@ -3,17 +3,22 @@ from pyterrier_doc2query import Doc2Query
 
 doc2query = Doc2Query()
 
-def predict(input_df):
-  return input_df
+def predict(input):
+  return doc2query(input)
 
 gr.Interface(
     predict,
     inputs=gr.Dataframe(
-            headers=["docno", "text"],
-            datatype=["str", "str"],
-            row_count=5,
-            col_count=(2, "fixed"),
-        ),
-    outputs="dataframe",
+      headers=["docno", "text"],
+      datatype=["str", "str"],
+      col_count=(2, "fixed"),
+      value=[['0', 'The presence of communication amid scientific minds was equally important to the success of the Manhattan Project as scientific intellect was. The only cloud hanging over the impressive achievement of the atomic researchers and engineers is what their success truly meant; hundreds of thousands of innocent lives obliterated.']],
+    ),
+    outputs=gr.Dataframe(
+      headers=["docno", "text", "querygen"],
+      datatype=["str", "str", "str"],
+      row_count=0,
+      col_count=0,
+    ),
     title="pyterrier_doc2query",
 ).launch()