Update app.py

app.py

@@ -31,12 +31,12 @@ login(token = os.getenv('gemma'))
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("image-text-to-text", model="google/gemma-3-4b-it")
+pipe = pipeline("image-text-to-text", model="google/gemma-3-4b-it").to("cuda")
 print(pipe(messages))
 
 
 ckpt = "google/gemma-3-4b-it"
-model = Gemma3ForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,)
+model = Gemma3ForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt)
 
 @spaces.GPU