Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
|
|
4 |
import torch
|
5 |
|
6 |
# Load model and processor
|
7 |
-
model_id = "cosmo3769/
|
8 |
model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
|
9 |
processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
|
10 |
|
|
|
4 |
import torch
|
5 |
|
6 |
# Load model and processor
|
7 |
+
model_id = "cosmo3769/finetuned_paligemma_vqav2_small_test1"
|
8 |
model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
|
9 |
processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
|
10 |
|