fix bug

.ipynb_checkpoints/app-checkpoint.py

@@ -18,7 +18,7 @@ def infer(s1, s2):
     if "error" in data:
         return "Error: "+ data["error"]
     elif "generated_text" in data[0]:
-        return data["generated_text"]
+        return data[0]["generated_text"]
     else:
         return data
 
@@ -26,20 +26,20 @@ 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."
 
-example1 = ["On Monday, Tom went to the market.","Tom went to the market on Monday."]
-
-s1 = gr.Textbox(value="One",label="Sentence 1")
-s2 = gr.Textbox(value="Two",label="Sentence 2")
+s1 = gr.Textbox(value="On Monday, Tom went to the market.",label="Sentence 1")
+s2 = gr.Textbox(value="Tom went to the market.",label="Sentence 2")
 
 examples = gr.Examples(
         examples=[
             ["one", "two"],
         ],
         inputs=[s1, s2],
-    )
+)
 
 demo = gr.Interface(fn=infer, inputs=[s1,s2], outputs="text",
                     title=title,
                     description=desc,
-                    article=long_desc)
+                    article=long_desc,
+                    )
+
 demo.launch()

app.py

@@ -18,7 +18,7 @@ def infer(s1, s2):
     if "error" in data:
         return "Error: "+ data["error"]
     elif "generated_text" in data[0]:
-        return data["generated_text"]
+        return data[0]["generated_text"]
     else:
         return data
 
@@ -26,20 +26,20 @@ 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."
 
-example1 = ["On Monday, Tom went to the market.","Tom went to the market on Monday."]
-
-s1 = gr.Textbox(value="One",label="Sentence 1")
-s2 = gr.Textbox(value="Two",label="Sentence 2")
+s1 = gr.Textbox(value="On Monday, Tom went to the market.",label="Sentence 1")
+s2 = gr.Textbox(value="Tom went to the market.",label="Sentence 2")
 
 examples = gr.Examples(
         examples=[
             ["one", "two"],
         ],
         inputs=[s1, s2],
-    )
+)
 
 demo = gr.Interface(fn=infer, inputs=[s1,s2], outputs="text",
                     title=title,
                     description=desc,
-                    article=long_desc)
+                    article=long_desc,
+                    )
+
 demo.launch()