Update app.py

app.py

@@ -43,6 +43,8 @@ model_m = AutoModel.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.float16
 ).to(device).eval()
+# Fix AssertionError by setting img_context_token_id
+model_m.img_context_token_id = tokenizer_m.convert_tokens_to_ids("<image>")
 
 # Load Space Thinker
 MODEL_ID_Z = "remyxai/SpaceThinker-Qwen2.5VL-3B"
@@ -93,7 +95,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
         if image is None:
             yield "Please upload an image."
             return
-        # Construct message with <image> token as per reference
+        # Construct message with <image> token
         if "<image>" not in text:
             message = f"<image>\n{text}"
         else:
@@ -346,7 +348,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             model_choice = gr.Radio(
                 choices=["Llama-3.1-Nemotron-Nano-VL-8B-V1", "SpaceThinker-3B", "coreOCR-7B-050325-preview"],
                 label="Select Model",
-                value="Llama-3.1-Nemotron-Nano-VL-8B-V1"  # Updated default value to a valid choice
+                value="Llama-3.1-Nemotron-Nano-VL-8B-V1"
             )
             
             gr.Markdown("**Model Info**")