Update main.py
Browse files
main.py
CHANGED
@@ -15,7 +15,7 @@ app = FastAPI()
|
|
15 |
|
16 |
# Model and class information
|
17 |
model_path = "model.keras"
|
18 |
-
class_indices = {0: '
|
19 |
|
20 |
# Load the model if it exists
|
21 |
if os.path.exists(model_path):
|
|
|
15 |
|
16 |
# Model and class information
|
17 |
model_path = "model.keras"
|
18 |
+
class_indices = {0: 'blight', 1: 'brown_spots'}
|
19 |
|
20 |
# Load the model if it exists
|
21 |
if os.path.exists(model_path):
|