Hugging Face's logo

Spaces:
onuralpszr
/
paligemma2-detection
Running on Zero

App Files Community
onuralpszr commited on
Commit
7aa8e30
·
verified ·
1 Parent(s): cba65f8

chore: extra variable removed

Browse files

Signed-off-by: Onuralp SEZER <[email protected]>

Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -15,9 +15,7 @@ LABEL_ANNOTATOR = sv.LabelAnnotator()
15
  MASK_ANNOTATOR = sv.MaskAnnotator()
16
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
 
18
-
19
  model_id = "google/paligemma2-3b-pt-448"
20
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(DEVICE)
22
  processor = PaliGemmaProcessor.from_pretrained(model_id)
23
 
 
15
  MASK_ANNOTATOR = sv.MaskAnnotator()
16
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
 
 
18
  model_id = "google/paligemma2-3b-pt-448"
 
19
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(DEVICE)
20
  processor = PaliGemmaProcessor.from_pretrained(model_id)
21