k23064919 commited on
Commit
a7c1326
·
1 Parent(s): aefbb6d

quickfix intermediate cnn clearml task id and defualt model loading in app.py

Browse files
Files changed (2) hide show
  1. ui/app.py +1 -0
  2. 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": "a517615e998c40c3a9957ecd02594010"
14
  }
15
  }
 
10
  "Intermediate CNN": {
11
  "description": "modelOne trained on 10 epochs",
12
  "model_type" : "cnn",
13
+ "clearml_task_id": "468137833d624b98a7b7c472275942a5"
14
  }
15
  }