made model names better
Browse files
app.py
CHANGED
@@ -91,6 +91,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
|
|
91 |
The image must be a numpy array of shape (C, W, W) or (1, C, W, W)
|
92 |
'''
|
93 |
|
|
|
94 |
num_channels = int(num_channels)
|
95 |
W = int(dim)
|
96 |
|
@@ -180,7 +181,7 @@ if __name__ == "__main__":
|
|
180 |
|
181 |
demo = gr.Interface(
|
182 |
fn=predict_and_analyze,
|
183 |
-
inputs=[gr.Dropdown(["
|
184 |
value="efficientnet",
|
185 |
label="Model Selection",
|
186 |
show_label=True),
|
|
|
91 |
The image must be a numpy array of shape (C, W, W) or (1, C, W, W)
|
92 |
'''
|
93 |
|
94 |
+
model_name = model_name.lower()
|
95 |
num_channels = int(num_channels)
|
96 |
W = int(dim)
|
97 |
|
|
|
181 |
|
182 |
demo = gr.Interface(
|
183 |
fn=predict_and_analyze,
|
184 |
+
inputs=[gr.Dropdown(["EfficientNet", "RegNet"],
|
185 |
value="efficientnet",
|
186 |
label="Model Selection",
|
187 |
show_label=True),
|