k23064919 commited on
Commit
588037c
·
1 Parent(s): 7acdd63

remove wrongly inserted variable assignment

Browse files
Files changed (1) hide show
  1. ui/app.py +1 -2
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, self.class_names = self.model_loader.loadModel(modelName)
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