double

app.py

@@ -170,14 +170,8 @@ def save_model(model, path):
     torch.save(model.state_dict(), path)
 
 
-def load_model(filename):
+def load_model(path):
     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)
@@ -440,15 +434,8 @@ def generate_labels(font_file=None, num_labels: int = 25):
         return [error_img]
 
 def list_fonts():
-    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
-
+    fonts = [f for f in os.listdir() if f.lower().endswith((".ttf", ".otf"))]
+    return ["None"] + fonts if fonts else ["None"]
 
 custom_css = """
 #label-gallery .gallery-item img {

models/Glipervelz.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:003cba654cea4a725f1eb920a58a3e43d088c23069b0f22f8f11807c9e68b3fa
-size 4388527