Spaces:
Running
Running
updated app.py
Browse files
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=(
|
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
|