fix bug

.ipynb_checkpoints/app-checkpoint.py

@@ -17,14 +17,14 @@ def infer(s1, s2):
 
     if "error" in data:
         return "Error: "+ data["error"]
+    elif "generated_text" in data:
+        return data["generated_text"]
     else:
         return data
 
-gr.Markdown("""
-# Paraphrase Classification and Explanation
-
-This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation
-""")
+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")
 

app.py

@@ -17,14 +17,14 @@ def infer(s1, s2):
 
     if "error" in data:
         return "Error: "+ data["error"]
+    elif "generated_text" in data:
+        return data["generated_text"]
     else:
         return data
 
-gr.Markdown("""
-# Paraphrase Classification and Explanation
-
-This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation
-""")
+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")