breadlicker45 commited on
Commit
9174c87
·
verified ·
1 Parent(s): 9427bdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def load_model():
23
  )
24
 
25
  # Load the processor and model using the correct identifier
26
- model_id = "google/paligemma2-10b-pt-896"
27
  processor = PaliGemmaProcessor.from_pretrained(model_id, use_auth_token=token)
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
  model = PaliGemmaForConditionalGeneration.from_pretrained(
 
23
  )
24
 
25
  # Load the processor and model using the correct identifier
26
+ model_id = "google/paligemma2-10b-pt-448"
27
  processor = PaliGemmaProcessor.from_pretrained(model_id, use_auth_token=token)
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
  model = PaliGemmaForConditionalGeneration.from_pretrained(