Spaces:
Sleeping
Sleeping
quickfix intermediate cnn clearml task id and defualt model loading in app.py
Browse files- ui/app.py +1 -0
- ui/config.py +1 -1
ui/app.py
CHANGED
|
@@ -23,6 +23,7 @@ from config import *
|
|
| 23 |
class PlantDiseaseApp:
|
| 24 |
def __init__(self):
|
| 25 |
self.model_loader = ModelLoader()
|
|
|
|
| 26 |
self.current_modelName = "Basic CNN"
|
| 27 |
self.model = self.model_loader.loadModel(self.current_modelName)
|
| 28 |
self.flagged_predictions = []
|
|
|
|
| 23 |
class PlantDiseaseApp:
|
| 24 |
def __init__(self):
|
| 25 |
self.model_loader = ModelLoader()
|
| 26 |
+
self.current_modelName = 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 = []
|
ui/config.py
CHANGED
|
@@ -10,6 +10,6 @@ MODEL_CONFIGS = {
|
|
| 10 |
"Intermediate CNN": {
|
| 11 |
"description": "modelOne trained on 10 epochs",
|
| 12 |
"model_type" : "cnn",
|
| 13 |
-
"clearml_task_id": "
|
| 14 |
}
|
| 15 |
}
|
|
|
|
| 10 |
"Intermediate CNN": {
|
| 11 |
"description": "modelOne trained on 10 epochs",
|
| 12 |
"model_type" : "cnn",
|
| 13 |
+
"clearml_task_id": "468137833d624b98a7b7c472275942a5"
|
| 14 |
}
|
| 15 |
}
|