Add error handle for loading model

app.py

@@ -118,8 +118,12 @@ def query(payload):
 
 def predict(text):
     print(preprocess(text))
-    output = query(preprocess(text))[0]
-    print(output)
+    
+    try:
+        output = query(preprocess(text))[0]
+    except Exception:
+        print(output)
+
     output = [tuple(i.values()) for i in output]
     output = dict((x, y) for x, y in output)