Update app.py

app.py

@@ -45,17 +45,15 @@ def initialize_model(model_file):
 
 def get_text(img_org):
     try:
-        print("debugger1")
+        # img_org = Image.open(image_path)
+        # Preprocess. Model expects a batch of images with shape: (B, C, H, W)
         x = transform(img_org.convert('RGB')).unsqueeze(0)
-        print("debugger2")
+    
+        # compute ONNX Runtime output prediction
         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, probs = tokenizer_base.decode(probs)
         preds = preds[0]
         print(preds)
         return preds