Spaces:
Sleeping
Sleeping
liewchooichin
commited on
Commit
•
e35d994
1
Parent(s):
0c08748
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"
|