Ling Lin

app.py

@@ -170,8 +170,14 @@ def save_model(model, path):
     torch.save(model.state_dict(), path)
 
 
-def load_model(path):
+def load_model(filename):
     global ocr_model
+    model_dir = "./models"
+    path = os.path.join(model_dir, filename)
+    
+    if not os.path.exists(path):
+        return f"Model file '{path}' does not exist."
+
     model = OCRModel(num_classes=len(CHAR2IDX))
     model.load_state_dict(torch.load(path, map_location=device))
     model.to(device)
@@ -434,8 +440,15 @@ def generate_labels(font_file=None, num_labels: int = 25):
         return [error_img]
 
 def list_fonts():
-    fonts = [f for f in os.listdir() if f.lower().endswith((".ttf", ".otf"))]
-    return ["None"] + fonts if fonts else ["None"]
+    font_dir = "./fonts"
+    if not os.path.exists(font_dir):
+        return ["None"]
+    fonts = [
+        (f, os.path.join(font_dir, f)) for f in os.listdir(font_dir)
+        if f.lower().endswith((".ttf", ".otf"))
+    ]
+    return [("None", "None")] + fonts
+
 
 custom_css = """
 #label-gallery .gallery-item img {