breadlicker45 commited on
Commit
d61ac05
·
verified ·
1 Parent(s): 38a09e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,10 +18,10 @@ def load_model():
18
 
19
  # Load the processor and model using the correct identifier
20
  processor = AutoProcessor.from_pretrained(
21
- "google/paligemma2-3b-pt-224", use_auth_token=token
22
  )
23
  model = AutoModelForImageTextToText.from_pretrained(
24
- "google/paligemma2-3b-pt-224", use_auth_token=token
25
  )
26
 
27
  return processor, model
 
18
 
19
  # Load the processor and model using the correct identifier
20
  processor = AutoProcessor.from_pretrained(
21
+ "google/paligemma2-28b-pt-896", use_auth_token=token
22
  )
23
  model = AutoModelForImageTextToText.from_pretrained(
24
+ "google/paligemma2-28b-pt-896", use_auth_token=token
25
  )
26
 
27
  return processor, model