Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -153,7 +153,7 @@ def create_tags(image, threshold):
|
|
153 |
tensor = transform(img).unsqueeze(0)
|
154 |
|
155 |
with torch.no_grad():
|
156 |
-
probits = model(tensor)
|
157 |
indices = torch.where(probits > threshold)[0]
|
158 |
values = probits[indices]
|
159 |
|
|
|
153 |
tensor = transform(img).unsqueeze(0)
|
154 |
|
155 |
with torch.no_grad():
|
156 |
+
probits = model(tensor).squeeze()
|
157 |
indices = torch.where(probits > threshold)[0]
|
158 |
values = probits[indices]
|
159 |
|