updage

README.md

@@ -1,12 +1,15 @@
 ---
-title: Pyterrier Doc2query
-emoji: 📚
-colorFrom: red
-colorTo: gray
+title: PyTerrier Doc2Query
+emoji: 🐕
+colorFrom: #F25F5C
+colorTo: #6096BA
 sdk: gradio
 sdk_version: 3.7
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+This is a demonstration of [PyTerrier's Doc2Query package](https://github.com/terrierteam/pyterrier_doc2query). Doc2Query generates
+queries for a document, which can then be appended to a document's text before indexing to boost important terms and add missing terms.
+
+Try it below!

app.py

@@ -12,15 +12,16 @@ def df2code(df):
 {rows}
 ])'''
 
-def predict(input, append):
+def predict(input, append, num_samples):
   doc2query.append = append
+  doc2query.num_samples = num_samples
   code = f'''
 **Code:**
 
 ```python
 import pandas as pd
 from pyterrier_doc2query import Doc2Query
-doc2query = Doc2Query(append={append})
+doc2query = Doc2Query(append={append}, num_samples={num_samples})
 doc2query({df2code(input)})
 ```
 '''
@@ -39,6 +40,12 @@ gr.Interface(
     ), gr.Checkbox(
       value=False,
       label="Append",
+    ), gr.Slider(
+      minimum=1,
+      maximum=10,
+      value=3,
+      step=1.,
+      label='# Queries'
     )],
     outputs=[gr.Dataframe(
       headers=["docno", "text", "querygen"],
@@ -50,5 +57,6 @@ gr.Interface(
       value=[["[docno]", "[text]", "[querygen]"]],
     ), gr.Markdown()],
     title="PyTerrier: Doc2Query",
+    description=open('README.md', 'rt').read().split('---\n')[-1],
     allow_flagging='never',
 ).launch()