Spaces:

jonaschua
/

gemma

Running on Zero

jonaschua commited on 20 days ago

Commit

e8b3607

verified ·

1 Parent(s): 1b792af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ login(token = os.getenv('gemma'))
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("image-text-to-text", model="google/gemma-3-4b-it").to("cuda")
 print(pipe(messages))

 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
+pipe = pipeline("image-text-to-text", model="google/gemma-3-4b-it")
 print(pipe(messages))