Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ def predict(img):
|
|
| 16 |
img_array = img_array / 255.0 # normalization
|
| 17 |
|
| 18 |
prediction = model.predict(img_array)[0][0]
|
| 19 |
-
label = "
|
| 20 |
return label
|
| 21 |
|
| 22 |
gr.Interface(
|
|
|
|
| 16 |
img_array = img_array / 255.0 # normalization
|
| 17 |
|
| 18 |
prediction = model.predict(img_array)[0][0]
|
| 19 |
+
label = "Uninfected" if prediction > 0.5 else "Infected"
|
| 20 |
return label
|
| 21 |
|
| 22 |
gr.Interface(
|