fix bug

.ipynb_checkpoints/app-checkpoint.py

@@ -14,9 +14,13 @@ def query(payload):
 def infer(s1, s2):
     model_input = "Classify and explain the relationship between this pair of sentences: <S1> "+s1+" </S1><S2> "+s2+" </S2>"
     data = query(model_input)
-    return data
 
-gradio.Markdown("""
+    if "error" in data:
+        return "Error: "+ data["error"]
+    else:
+        return data
+
+gr.Markdown("""
 # Paraphrase Classification and Explanation
 
 This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation

app.py

@@ -14,9 +14,13 @@ def query(payload):
 def infer(s1, s2):
     model_input = "Classify and explain the relationship between this pair of sentences: <S1> "+s1+" </S1><S2> "+s2+" </S2>"
     data = query(model_input)
-    return data
 
-gradio.Markdown("""
+    if "error" in data:
+        return "Error: "+ data["error"]
+    else:
+        return data
+
+gr.Markdown("""
 # Paraphrase Classification and Explanation
 
 This is a Flan-T5-Large model fine-tuned to perform paraphrase classification and explanation