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

quickfix model name

Browse files
Files changed (1) hide show
  1. ui/app.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"
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 = "best_model.pth"
23
  self.model = self.model_loader.loadModel(self.current_modelName)
24
  self.flagged_predictions = []
25