Spaces:
Sleeping
Sleeping
quickfix model names
Browse files
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
|
| 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
|
| 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
|
| 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
|