Spaces:
Build error
Build error
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),
|