Update app.py

app.py

@@ -190,16 +190,6 @@ model_d = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True
 ).eval()
 
-# Load LightOnOCR-1B-1025
-MODEL_PATH_L = "lightonai/LightOnOCR-1B-1025"
-processor_l = AutoProcessor.from_pretrained(MODEL_PATH_L, trust_remote_code=True)
-model_l = AutoModelForImageTextToText.from_pretrained(
-    MODEL_PATH_L,
-    torch_dtype=torch.bfloat16,
-    device_map="auto",
-    trust_remote_code=True
-).eval()
-
 # Load olmOCR-2-7B-1025
 MODEL_ID_M = "allenai/olmOCR-2-7B-1025"
 processor_m = AutoProcessor.from_pretrained(MODEL_ID_M, trust_remote_code=True)
@@ -229,9 +219,6 @@ def generate_image(model_name: str, text: str, image: Image.Image,
     elif model_name == "Dots.OCR":
         processor = processor_d
         model = model_d
-    elif model_name == "LightOnOCR":
-        processor = processor_l
-        model = model_l
     else:
         yield "Invalid model selected.", "Invalid model selected."
         return
@@ -308,7 +295,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                     markdown_output = gr.Markdown(label="(Result.Md)")
 
                 model_choice = gr.Radio(
-                    choices=["Nanonets-OCR2-3B", "Chandra-OCR", "olmOCR-2-7B-1025", "Dots.OCR", "LightOnOCR"],
+                    choices=["Nanonets-OCR2-3B", "Chandra-OCR", "olmOCR-2-7B-1025", "Dots.OCR"],
                     label="Select Model",
                     value="Nanonets-OCR2-3B"
                 )