Omar-Mostafa
commited on
Commit
•
e6d7387
1
Parent(s):
0db56ea
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def predict(img):
|
|
38 |
# confidences = {"Cat": prediction, 'Dog': 1 - prediction}
|
39 |
|
40 |
# Return a tuple with the predicted label and confidence
|
41 |
-
if prediction >= 0.
|
42 |
# return {'Dog': prediction, "Cat": 1-prediction}
|
43 |
return {'Dog': 1}
|
44 |
|
|
|
38 |
# confidences = {"Cat": prediction, 'Dog': 1 - prediction}
|
39 |
|
40 |
# Return a tuple with the predicted label and confidence
|
41 |
+
if prediction >= 0.49:
|
42 |
# return {'Dog': prediction, "Cat": 1-prediction}
|
43 |
return {'Dog': 1}
|
44 |
|