Samarth991 commited on
Commit
8753c71
1 Parent(s): 90b20c2

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
35
  #preds = outputs.logits.unsqueeze(1)
36
  preds = nn.functional.interpolate(
37
  outputs.logits.unsqueeze(1),
38
- size=(test_image.shape[0], test_image.shape[1]),
39
  mode="bilinear"
40
  )
41
  threshold = 0.1
 
35
  #preds = outputs.logits.unsqueeze(1)
36
  preds = nn.functional.interpolate(
37
  outputs.logits.unsqueeze(1),
38
+ size=(image.shape[0], image.shape[1]),
39
  mode="bilinear"
40
  )
41
  threshold = 0.1