cosmo3769 commited on
Commit
f41e04e
1 Parent(s): eadc99d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  import torch
5
 
6
  # Load model and processor
7
- model_id = "cosmo3769/finetuned_paligemma_vqav2_small"
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