Spaces:
Build error
Build error
LayBraid
commited on
Commit
·
21884ee
1
Parent(s):
565dc8e
add requirements.txt
Browse files
app.py
CHANGED
|
@@ -40,6 +40,7 @@ def send_inputs(img):
|
|
| 40 |
outputs = model(**inputs)
|
| 41 |
logits_per_image = outputs.logits_per_image
|
| 42 |
probs = logits_per_image.softmax(dim=1)
|
|
|
|
| 43 |
return probs
|
| 44 |
|
| 45 |
|
|
|
|
| 40 |
outputs = model(**inputs)
|
| 41 |
logits_per_image = outputs.logits_per_image
|
| 42 |
probs = logits_per_image.softmax(dim=1)
|
| 43 |
+
print(probs)
|
| 44 |
return probs
|
| 45 |
|
| 46 |
|