Update app.py

app.py

@@ -44,14 +44,23 @@ def initialize_model(model_file):
     return transform,ort_session 
 
 def get_text(img_org):
-    x = transform(img_org.convert('RGB')).unsqueeze(0)
-    ort_inputs = {ort_session.get_inputs()[0].name: to_numpy(x)}
-    logits = ort_session.run(None, ort_inputs)[0]
-    probs = torch.tensor(logits).softmax(-1)
-    preds, _ = tokenizer_base.decode(probs)
-    preds = preds[0]
-    print(preds)
-    return preds
+    try:
+        print("debugger1")
+        x = transform(img_org.convert('RGB')).unsqueeze(0)
+        print("debugger2")
+        ort_inputs = {ort_session.get_inputs()[0].name: to_numpy(x)}
+        print("debugger3")
+        logits = ort_session.run(None, ort_inputs)[0]
+        print("debugger4")
+        probs = torch.tensor(logits).softmax(-1)
+        print("debugger5")
+        preds, _ = tokenizer_base.decode(probs)
+        print("debugger6")
+        preds = preds[0]
+        print(preds)
+        return preds
+    except Exception as e:
+        print(str(e))
 
 transform,ort_session = initialize_model(model_file=model_file)