Update app.py

app.py

@@ -4,6 +4,13 @@ from inductor import BartInductor
 inductor = BartInductor()
 
 def bart(text):
+    results = inductor.generate(text)
+    #results=[('<mask> is the capital and largest city of <mask>.', 0.0075895977795922415), ('<mask> is the largest city in <mask>.', 0.005302619071099926), ('<mask> is the most populous state in <mask>.', 0.0036559513972044544), ('<mask> is the capital of <mask>.', 0.0028800265140167813), ('<mask> is a state in <mask>.', 0.002295685033583364), ('<mask> is a capital of <mask>.', 0.001348920570936385), ('<mask> has one of the highest rates of poverty in <mask>.', 0.0012284877234430835), ('<mask> is a major commercial and financial centre of <mask>.', 0.001225910195650215), ('<mask> was then a part of <mask>.', 0.0007620440367924856), ('<mask>, the capital of the country, is the largest city in <mask>.', 0.0006940517028590776)]
+    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
+
+def bart2(text):
     #results = inductor.generate(text)
     results=[('<mask> is the capital and largest city of <mask>.', 0.0075895977795922415), ('<mask> is the largest city in <mask>.', 0.005302619071099926), ('<mask> is the most populous state in <mask>.', 0.0036559513972044544), ('<mask> is the capital of <mask>.', 0.0028800265140167813), ('<mask> is a state in <mask>.', 0.002295685033583364), ('<mask> is a capital of <mask>.', 0.001348920570936385), ('<mask> has one of the highest rates of poverty in <mask>.', 0.0012284877234430835), ('<mask> is a major commercial and financial centre of <mask>.', 0.001225910195650215), ('<mask> was then a part of <mask>.', 0.0007620440367924856), ('<mask>, the capital of the country, is the largest city in <mask>.', 0.0006940517028590776)]
     results = [(result[0], float(result[1]) * 100) for result in results]
@@ -13,15 +20,15 @@ def bart(text):
 bivar = gr.Interface(fn=bart, 
                      inputs=gr.inputs.Textbox(default='<mask> is the capital of <mask>.'),
                      outputs=gr.Label(),
-                     title="BART Inductor",
-                     examples=[['<mask> is the capital of <mask>.']],
+                     title="BART Inductor 双变量",
+                     examples=[['<mask> is the capital of <mask>.'],['<mask> is founder of <mask>.']],
                      description="Enter a text prompt to generate text using BART.")
 
-trivar = gr.Interface(fn=bart, 
-                     inputs=gr.inputs.Textbox(default='<mask> is the capital of <mask>.'),
+trivar = gr.Interface(fn=bart2, 
+                     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 capital of <mask>.']],
+                     title="BART Inductor 三变量",
+                     examples=[['<mask> is the parent of <mask>, who is the sibling of <mask>.']],
                      description="Enter a text prompt to generate text using BART.")
 
 demo = gr.TabbedInterface([bivar, trivar], ["双变量", "三变量"])