Update app.py

app.py

@@ -40,7 +40,7 @@ def get_width(image, height=DEFAULT_HEIGHT):
     return height * aspect_ratio
 
 def predict(model_name, input_img):
-    model_dir = 'pylaia-mcdonald_v2'
+    model_dir = 'pylaia-samaritan_v1'
     temperature = 2.0
     batch_size = 1
 
@@ -118,7 +118,7 @@ def process_image(image):
         crop_pil = Image.fromarray(cv2.cvtColor(crop_img, cv2.COLOR_BGR2RGB))
 
         # Recognize text using PyLaia model
-        predicted = predict('johnlockejrr/pylaia-samaritan_v1', crop_pil)
+        predicted = predict('pylaia-samaritan_v1', crop_pil)
         texts.append(predicted[1]["text"])
 
         bboxes.append((x1, y1, x2, y2))