Update app.py

app.py

@@ -11,9 +11,13 @@ def func(query):
             output = model.forward(number=100)["output"]
             return output
     except Exception as e:
-        return str(e)
+        return "Error [A]" + str(e)
     else:
-        return generate(query)
+        try:
+            a = generate(query)
+            return a
+        except Exception as e:
+            return "Error [B]" + str(e)
 
 iface = gr.Interface(fn=func, inputs="text", outputs="text")
 iface.launch()