k23064919 commited on
Commit
0791b11
·
1 Parent(s): 1fc1df9

quickfix model name

Browse files
Files changed (2) hide show
  1. ui/app.py +1 -1
  2. ui/config.py +1 -1
ui/app.py CHANGED
@@ -19,7 +19,7 @@ from model_loader import ModelLoader
19
  class PlantDiseaseApp:
20
  def __init__(self):
21
  self.model_loader = ModelLoader()
22
- self.current_modelName = "best_model.pth"
23
  self.model = self.model_loader.loadModel(self.current_modelName)
24
  self.flagged_predictions = []
25
 
 
19
  class PlantDiseaseApp:
20
  def __init__(self):
21
  self.model_loader = ModelLoader()
22
+ self.current_modelName = "model_one.pt"
23
  self.model = self.model_loader.loadModel(self.current_modelName)
24
  self.flagged_predictions = []
25
 
ui/config.py CHANGED
@@ -5,6 +5,6 @@ MODEL_CONFIGS = {
5
  "CNN from Scratch": {
6
  "description": "Custom CNN model trained from scratch",
7
  "model_type": "cnn",
8
- "clearml_task_id": "fe14662da63d45bf9208fdf9856d2fcc"
9
  }
10
  }
 
5
  "CNN from Scratch": {
6
  "description": "Custom CNN model trained from scratch",
7
  "model_type": "cnn",
8
+ "clearml_task_id": "12922d47f3be4d14856a84cdabfe5cb5"
9
  }
10
  }