Akshayram1 commited on
Commit
72fe4af
·
verified ·
1 Parent(s): e1ce9ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from transformers import PaliGemmaForConditionalGeneration, PaliGemmaProcessor
6
 
7
  # Model and Processor Setup
8
- model_id = "gv-hf/paligemma2-3b-mix-448"
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
  HF_KEY = os.getenv("HF_KEY")
11
  if not HF_KEY:
 
5
  from transformers import PaliGemmaForConditionalGeneration, PaliGemmaProcessor
6
 
7
  # Model and Processor Setup
8
+ model_id = "google/paligemma2-3b-mix-448"
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
  HF_KEY = os.getenv("HF_KEY")
11
  if not HF_KEY: