Merge branch 'main' of hf.co:spaces/ongkn/attraction-classifier-subjective-v4
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def classify_image(input):
|
|
| 62 |
elif result[0]["label"] == "pos" and result[0]["score"] > 0.99:
|
| 63 |
return result[0]["label"], result[0]["score"], str("**ABSOLUTELY MINDBLOWING**"), face, dffImage, gradCamImage
|
| 64 |
else:
|
| 65 |
-
return
|
| 66 |
|
| 67 |
iface = gr.Interface(
|
| 68 |
fn=classify_image,
|
|
|
|
| 62 |
elif result[0]["label"] == "pos" and result[0]["score"] > 0.99:
|
| 63 |
return result[0]["label"], result[0]["score"], str("**ABSOLUTELY MINDBLOWING**"), face, dffImage, gradCamImage
|
| 64 |
else:
|
| 65 |
+
return cls, result[0]["score"], "Indifferent", face, dffImage, gradCamImage
|
| 66 |
|
| 67 |
iface = gr.Interface(
|
| 68 |
fn=classify_image,
|