merve HF staff commited on
Commit
8d0672c
·
1 Parent(s): 530cb47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def siglip_detector(image, texts):
22
  padding="max_length")
23
 
24
  with torch.no_grad():
25
- outputs = model(**inputs)
26
  logits_per_image = outputs.logits_per_image
27
  probs = torch.sigmoid(logits_per_image)
28
  return probs
 
22
  padding="max_length")
23
 
24
  with torch.no_grad():
25
+ outputs = siglip_model(**inputs)
26
  logits_per_image = outputs.logits_per_image
27
  probs = torch.sigmoid(logits_per_image)
28
  return probs