Update app.py

app.py

@@ -3,27 +3,21 @@ from inductor import BartInductor
 
 inductor = BartInductor()
 
-def bart(text):
-    results = inductor.generate(text, return_scores=True)
+def bart(prompt, num):
+    results = inductor.generate(prompt, k=num, topk=num, return_scores=True)
     results = [(result[0], float(result[1]) * 100) for result in results]
     results_dict = {result[0]: float(result[1]) for result in results}
     return results_dict
 
-bivar = gr.Interface(fn=bart, 
-                     inputs=gr.inputs.Textbox(default='<mask> is the capital of <mask>.'),
+demo = gr.Interface(fn=bart, 
+                     inputs=[gr.inputs.Textbox(default='<mask> is the capital of <mask>.'),gr.Slider(0, 10)],
                      outputs=gr.Label(),
-                     title="BART Inductor 双变量",
-                     examples=[['<mask> is the capital of <mask>.'],['<mask> is founder and CEO of <mask>.']],
+                     title="Orion",
+                     examples=[['<mask> is the capital of <mask>.',5],
+                               ['<mask> is founder and CEO of <mask>.',5],
+                              ["<mask>'s mother was a <mask>-based actress, <mask>.",5]],
                      description="Enter a text prompt to generate text using BART.")
 
-trivar = gr.Interface(fn=bart, 
-                     inputs=gr.inputs.Textbox(default='<mask> is the parent of <mask>, who is the sibling of <mask>.'),
-                     outputs=gr.Label(),
-                     title="BART Inductor 三变量",
-                     examples=[['<mask> is the parent of <mask>, who is the sibling of <mask>.']],
-                     description="this might not work so well")
-
-demo = gr.TabbedInterface([bivar, trivar], ["双变量", "三变量"])
 
 if __name__ == "__main__":
     demo.launch()