Update names

app.py

@@ -178,19 +178,19 @@ def generate_image(model_name: str, text: str, image: Image.Image,
     Generates responses using the selected model for image input.
     Yields raw text and Markdown-formatted text.
     """
-    if model_name == "RolmOCR-7B":
+    if model_name == "Model 3":
         processor = processor_m
         model = model_m
-    elif model_name == "Qwen2-VL-OCR-2B":
+    elif model_name == "Model 5":
         processor = processor_x
         model = model_x
-    elif model_name == "Nanonets-OCR2-3B":
+    elif model_name == "Model 1":
         processor = processor_v
         model = model_v
-    elif model_name == "Aya-Vision-8B":
+    elif model_name == "Model 4":
         processor = processor_a
         model = model_a
-    elif model_name == "olmOCR-7B-0725":
+    elif model_name == "Model 2":
         processor = processor_w
         model = model_w
     else:
@@ -247,7 +247,7 @@ image_examples = [
 
 # Create the Gradio Interface
 with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
-    gr.Markdown("# **Multimodal OCR**", elem_id="main-title")
+    gr.Markdown("# **RMA OCR**", elem_id="main-title")
     with gr.Row():
         with gr.Column(scale=2):
             image_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
@@ -273,10 +273,10 @@ 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", "olmOCR-7B-0725", "RolmOCR-7B",
-                         "Aya-Vision-8B", "Qwen2-VL-OCR-2B"],
+                    choices=["Model 1", "Model 2", "Model 3",
+                         "Model 4", "Model 5"],
                     label="Select Model",
-                    value="Nanonets-OCR2-3B"
+                    value="Model 1"
                 )
 
     image_submit.click(