update

app.py

@@ -1,3 +1,4 @@
+import base64
 import pandas as pd
 import gradio as gr
 from pyterrier_doc2query import Doc2Query
@@ -6,6 +7,12 @@ MODEL = 'macavaney/doc2query-t5-base-msmarco'
 
 doc2query = Doc2Query(MODEL, append=True, num_samples=5)
 
+COLAB_NAME = 'pyterrier_doc2query.ipynb'
+COLAB_INSTALL = '''
+!pip install -q git+https://github.com/terrier-org/pyterrier
+!pip install -q git+https://github.com/terrierteam/pyterrier_doc2query
+'''.strip()
+
 def df2code(df):
   rows = []
   for row in df.itertuples(index=False):
@@ -15,21 +22,32 @@ def df2code(df):
 {rows}
 ])'''
 
+def code2colab(code):
+  enc_code = base64.b64encode((COLAB_INSTALL + '\n\n' + code).encode()).decode()
+  url = f'https://colaburl.macavaney.us/?py64={enc_code}&name={COLAB_NAME}'
+  return f'<a href="{url}" rel="nofollow" target="_blank" style="float: right;"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" style="margin: 0;" /></a>'
+
+def code2md(code):
+  return f'''
+{code2colab(code)}
+
+**Code:**
+
+```python
+{code}
+```
+'''
+
 def predict(input, model, append, num_samples):
   assert model == MODEL
   doc2query.append = append
   doc2query.num_samples = num_samples
-  code = f'''
-**Code:**
-
-```python
-import pandas as pd
+  code = f'''import pandas as pd
 from pyterrier_doc2query import Doc2Query
 doc2query = Doc2Query({repr(model)}, append={append}, num_samples={num_samples})
 doc2query({df2code(input)})
-```
 '''
-  return (doc2query(input), code)
+  return (doc2query(input), code2md(code))
 
 example_inp = pd.DataFrame([
   {'docno': '0', 'text': '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.'},