Spaces:
Runtime error
Runtime error
remove wrongly inserted variable assignment
Browse files
ui/app.py
CHANGED
|
@@ -24,7 +24,6 @@ class PlantDiseaseApp:
|
|
| 24 |
def __init__(self):
|
| 25 |
self.model_loader = ModelLoader()
|
| 26 |
self.current_modelName = list(config.MODEL_CONFIGS.keys())[0]
|
| 27 |
-
self.current_modelName = "Basic CNN"
|
| 28 |
self.model = self.model_loader.loadModel(self.current_modelName)
|
| 29 |
self.flagged_predictions = []
|
| 30 |
self.class_names = utils.get_class_names()
|
|
@@ -49,7 +48,7 @@ class PlantDiseaseApp:
|
|
| 49 |
try:
|
| 50 |
# Load model if needed
|
| 51 |
if modelName != self.current_modelName:
|
| 52 |
-
self.model
|
| 53 |
self.current_modelName = modelName
|
| 54 |
|
| 55 |
# Preprocess image
|
|
|
|
| 24 |
def __init__(self):
|
| 25 |
self.model_loader = ModelLoader()
|
| 26 |
self.current_modelName = list(config.MODEL_CONFIGS.keys())[0]
|
|
|
|
| 27 |
self.model = self.model_loader.loadModel(self.current_modelName)
|
| 28 |
self.flagged_predictions = []
|
| 29 |
self.class_names = utils.get_class_names()
|
|
|
|
| 48 |
try:
|
| 49 |
# Load model if needed
|
| 50 |
if modelName != self.current_modelName:
|
| 51 |
+
self.model = self.model_loader.loadModel(modelName)
|
| 52 |
self.current_modelName = modelName
|
| 53 |
|
| 54 |
# Preprocess image
|