Spaces:
Sleeping
Sleeping
Update app.py
#1
by
jungukhur
- opened
app.py
CHANGED
|
@@ -62,7 +62,7 @@ transform = transforms.Compose([
|
|
| 62 |
])
|
| 63 |
|
| 64 |
# Class labels
|
| 65 |
-
class_names = [
|
| 66 |
|
| 67 |
# --------- Prediction Function ----------
|
| 68 |
def predict(image):
|
|
|
|
| 62 |
])
|
| 63 |
|
| 64 |
# Class labels
|
| 65 |
+
class_names = ['No Impairment', 'Very Mild Impairment', 'Moderate Impairment', 'Mild Impairment']
|
| 66 |
|
| 67 |
# --------- Prediction Function ----------
|
| 68 |
def predict(image):
|