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
|