Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def predict(img):
|
|
19 |
|
20 |
outputs = nn.functional.sigmoid(logits)
|
21 |
|
22 |
-
return outputs[0][0], outputs[0][1]
|
23 |
|
24 |
gr.Interface(
|
25 |
title="Artwork scorer",
|
|
|
19 |
|
20 |
outputs = nn.functional.sigmoid(logits)
|
21 |
|
22 |
+
return outputs[0][0].item(), outputs[0][1].item()
|
23 |
|
24 |
gr.Interface(
|
25 |
title="Artwork scorer",
|