fix bug

.ipynb_checkpoints/app-checkpoint.py

@@ -17,7 +17,7 @@ def infer(s1, s2):
 
     if "error" in data:
         return "Error: "+ data["error"]
-    elif "generated_text" in data:
+    elif "generated_text" in data[0]:
         return data["generated_text"]
     else:
         return data
@@ -26,6 +26,11 @@ title = "Paraphrase Classification and Explanation"
 desc = "Classify and explain the semantic relationship between the two sentences"
 long_desc = "This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation. It takes in two sentences as inputs."
 
-demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text")
+example1 = ["On Monday, Tom went to the market.","Tom went to the market on Monday."]
 
+demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text",
+                    examples=[example1,]
+                    title=title,
+                    description=desc,
+                    article=long_desc)
 demo.launch()

app.py

@@ -17,7 +17,7 @@ def infer(s1, s2):
 
     if "error" in data:
         return "Error: "+ data["error"]
-    elif "generated_text" in data:
+    elif "generated_text" in data[0]:
         return data["generated_text"]
     else:
         return data
@@ -26,6 +26,11 @@ title = "Paraphrase Classification and Explanation"
 desc = "Classify and explain the semantic relationship between the two sentences"
 long_desc = "This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation. It takes in two sentences as inputs."
 
-demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text")
+example1 = ["On Monday, Tom went to the market.","Tom went to the market on Monday."]
 
+demo = gr.Interface(fn=infer, inputs=["text", "text"], outputs="text",
+                    examples=[example1,]
+                    title=title,
+                    description=desc,
+                    article=long_desc)
 demo.launch()