Spaces:

Akshayram1
/

paligemma_2_demo_image_text_to_text

Running

Akshayram1 commited on Mar 3

Commit

0b73e4f

verified ·

1 Parent(s): 6fcae89

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ import jax.numpy as jnp
 import numpy as np
 import spaces
-model_id = "gv-hf/paligemma2-3b-mix-448"
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 HF_KEY = os.getenv("HF_KEY")

 import numpy as np
 import spaces
+model_id = "google/paligemma2-3b-mix-448"
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 HF_KEY = os.getenv("HF_KEY")