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 |
|