Update app.py

app.py

@@ -14,7 +14,7 @@ from loguru import logger
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForImageTextToText, TextIteratorStreamer
 
-model_id = os.getenv("MODEL_ID", "google/medgemma-4b-it")
+model_id = os.getenv("MODEL_ID", "google/medgemma-1.5-4b-it")
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForImageTextToText.from_pretrained(
     model_id, device_map="auto", torch_dtype=torch.bfloat16
@@ -206,7 +206,7 @@ def run(message: dict, history: list[dict], system_prompt: str = "", max_new_tok
 
 
 DESCRIPTION = """\
-This is a demo of MedGemma, a Gemma 3 variant trained for performance on medical text and image comprehension.
+This is a demo of MedGemma 1.5, a Gemma 3 variant trained for performance on medical text and image comprehension.
 You can upload images, interleaved images and videos. Note that video input only supports single-turn conversation and mp4 input.
 """
 
@@ -221,7 +221,7 @@ demo = gr.ChatInterface(
         gr.Slider(label="Max New Tokens", minimum=100, maximum=8192, step=10, value=2048),
     ],
     stop_btn=True,
-    title="MedGemma 4B IT",
+    title="MedGemma 1.5 4B IT",
     description=DESCRIPTION,
     run_examples_on_click=False,
     cache_examples=False,