jpterry commited on
Commit
28af0c6
·
1 Parent(s): d404815

made model names better

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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(["efficientnet", "regnet"],
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),