Spaces:
Sleeping
Sleeping
threshold 0.4, multilabel text box hidden
Browse files
app.py
CHANGED
|
@@ -159,7 +159,7 @@ def get_prediction(img_path):
|
|
| 159 |
# binary label
|
| 160 |
pred_binary = keras_binary_model(img_array, training=False)
|
| 161 |
print(f"Keras binary label: {pred_binary}")
|
| 162 |
-
|
| 163 |
fake = "Fake"
|
| 164 |
else:
|
| 165 |
fake = "Real"
|
|
|
|
| 159 |
# binary label
|
| 160 |
pred_binary = keras_binary_model(img_array, training=False)
|
| 161 |
print(f"Keras binary label: {pred_binary}")
|
| 162 |
+
if pred_binary[0][0] > threshold:
|
| 163 |
fake = "Fake"
|
| 164 |
else:
|
| 165 |
fake = "Real"
|