update

app.py

@@ -5,6 +5,7 @@ pt.init()
 from pyterrier_gradio import Demo, MarkdownFile, interface, df2code, code2md, EX_Q
 
 retr = pt.TerrierRetrieve.from_dataset('vaswani', 'terrier_stemmed')
+sdm = pt.rewrite.SDM()
 
 COLAB_NAME = 'pyterrier_retrieve.ipynb'
 COLAB_INSTALL = '''
@@ -33,15 +34,29 @@ retr({df2code(input)})'''
   res['score'] = res['score'].map(lambda x: round(x, 2))
   return (res, code2md(code, COLAB_INSTALL, COLAB_NAME))
 
+def predict_sdm(input):
+  code = f'''import pandas as pd
+import pyterrier as pt ; pt.init()
+
+sdm = pt.rewrite.SDM()
+
+sdm({df2code(input)})
+'''
+  res = sdm(input)
+  return (res, code2md(code, COLAB_INSTALL, COLAB_NAME))
+
+Q = pd.DataFrame([
+      ['1', 'measurement of dielectric constant of liquids by the use of microwave techniques'],
+      ['2', 'mathematical analysis and design details of waveguide fed microwave radiations'],
+      ['3', 'use of digital computers in the design of band pass filters having given phase and attenuation characteristics'],
+], columns=['qid', 'query'])
+
+
 interface(
   MarkdownFile('README.md'),
   Demo(
     predict,
-    pd.DataFrame([
-      ['1', 'measurement of dielectric constant of liquids by the use of microwave techniques'],
-      ['2', 'mathematical analysis and design details of waveguide fed microwave radiations'],
-      ['3', 'use of digital computers in the design of band pass filters having given phase and attenuation characteristics'],
-    ], columns=['qid', 'query']),
+    Q,
     [
     gr.Dropdown(
       choices=['vaswani stemmed'],
@@ -64,5 +79,12 @@ interface(
     )],
     scale=2/3
   ),
+  MarkdownFile('sdm.md'),
+  Demo(
+    predict_sdm,
+    Q,
+    [],
+    scale=2/3
+  ),
   MarkdownFile('wrapup.md'),
 ).launch(share=False)