Update app.py

app.py

@@ -29,7 +29,7 @@ else:
 from transformers import BitsAndBytesConfig
 import torch
 
-model_variant = "27b-text-it"  # @param ["4b-it", "27b-it", "27b-text-it"]
+model_variant = "4b-it"  # @param ["4b-it", "27b-it", "27b-text-it"]
 model_id = f"google/medgemma-{model_variant}"
 
 use_quantization = True  # @param {type: "boolean"}
@@ -66,15 +66,6 @@ else:
 pipe.model.generation_config.do_sample = False
 
 
-if "text" in model_variant:
-    from transformers import AutoModelForCausalLM, AutoTokenizer
-    model = AutoModelForCausalLM.from_pretrained(model_id, **model_kwargs)
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-else:
-    from transformers import AutoModelForImageTextToText, AutoProcessor
-    model = AutoModelForImageTextToText.from_pretrained(model_id, **model_kwargs)
-    processor = AutoProcessor.from_pretrained(model_id)
-
 
 role_instruction = "You are an expert radiologist."
 if "27b" in model_variant and is_thinking: