Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
5c9b0ca
1
Parent(s):
1acc0cd
fajsdlk
Browse files
app.py
CHANGED
|
@@ -11,6 +11,7 @@ DESCRIPTION = "Tag images with E621 tags"
|
|
| 11 |
MODEL_ID = "estrogen/paligemma2-3b-e621-224"
|
| 12 |
|
| 13 |
model = PaliGemmaForConditionalGeneration.from_pretrained(MODEL_ID)
|
|
|
|
| 14 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
| 15 |
|
| 16 |
@spaces.GPU
|
|
|
|
| 11 |
MODEL_ID = "estrogen/paligemma2-3b-e621-224"
|
| 12 |
|
| 13 |
model = PaliGemmaForConditionalGeneration.from_pretrained(MODEL_ID)
|
| 14 |
+
model.to("cuda")
|
| 15 |
processor = AutoProcessor.from_pretrained(MODEL_ID)
|
| 16 |
|
| 17 |
@spaces.GPU
|