dwb2023 commited on
Commit
68148d0
·
verified ·
1 Parent(s): 872f68f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ login(token=hf_token, add_to_git_credential=True)
41
 
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
43
 
44
- model_id = "google/paligemma-3b-mix-448""
45
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
46
  processor = PaliGemmaProcessor.from_pretrained(model_id)
47
 
 
41
 
42
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
43
 
44
+ model_id = "google/paligemma-3b-mix-448"
45
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
46
  processor = PaliGemmaProcessor.from_pretrained(model_id)
47