Update app.py

app.py

@@ -35,7 +35,7 @@ login(token = os.getenv('gemma'))
 # print(pipe(messages))
 
 
-ckpt = "google/gemma-3-27b-it"
+ckpt = "google/gemma-3-4b-it"
 model = Gemma3ForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt)