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

quickfix model names

Browse files
Files changed (1) hide show
  1. ui/app.py +3 -3
ui/app.py CHANGED
@@ -23,7 +23,7 @@ from config import *
23
  class PlantDiseaseApp:
24
  def __init__(self):
25
  self.model_loader = ModelLoader()
26
- self.current_modelName = "CNN from Scratch"
27
  self.model = self.model_loader.loadModel(self.current_modelName)
28
  self.flagged_predictions = []
29
  self.class_names = utils.get_class_names()
@@ -155,7 +155,7 @@ def create_interface():
155
  with gr.Row():
156
  model_selector = gr.Dropdown(
157
  choices=list(config.MODEL_CONFIGS.keys()),
158
- value="CNN from Scratch",
159
  label="Select Model",
160
  info="Choose which model to use for predictions"
161
  )
@@ -258,7 +258,7 @@ def create_interface():
258
  across 39 different plant disease categories.
259
 
260
  ### Model Architecture
261
- - **CNN from Scratch**: Custom convolutional neural network
262
  - **Transfer Learning**: Fine-tuned ResNet18 (if available)
263
 
264
  ### Technology Stack
 
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 = []
29
  self.class_names = utils.get_class_names()
 
155
  with gr.Row():
156
  model_selector = gr.Dropdown(
157
  choices=list(config.MODEL_CONFIGS.keys()),
158
+ value="Basic CNN",
159
  label="Select Model",
160
  info="Choose which model to use for predictions"
161
  )
 
258
  across 39 different plant disease categories.
259
 
260
  ### Model Architecture
261
+ - **Basic CNN**: Custom convolutional neural network
262
  - **Transfer Learning**: Fine-tuned ResNet18 (if available)
263
 
264
  ### Technology Stack