Added progress bar

app.py

@@ -7,7 +7,8 @@ from ultralytics import YOLO
 from PIL import Image
 import torch
 
-def process(path):
+def process(path, progress = gr.Progress()):
+    progress(0, desc="Starting")
     LINE_MODEL_PATH = "Kansallisarkisto/multicentury-textline-detection"
     #OCR_MODEL_PATH = "Kansallisarkisto/multicentury-htr-model"
     OCR_MODEL_PATH = "microsoft/trocr-large-handwritten"
@@ -18,7 +19,7 @@ def process(path):
 
     # Open an image of handwritten text
     image = Image.open(path).convert("RGB")
-
+    progress(0, desc="Extracting Text Lines")
     try:
         # Load the trained line detection model
         cached_model_path = hf_hub_download(repo_id = LINE_MODEL_PATH, filename="lines_20240827.pt")
@@ -31,7 +32,7 @@ def process(path):
     boxes = results.boxes.xyxy
     indices = boxes[:,1].sort().indices
     boxes = boxes[indices]
-    for box in boxes:
+    for box in progress.tqdm(boxes, desc="Text Recognition"):
         #box = box + torch.tensor([-10,0, 10, 0])
         box = [tensor.item() for tensor in box]
         lineImg = image.crop(tuple(list(box)))