RohitGandikota commited on
Commit
0a3fdbd
1 Parent(s): e1dc863

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ class Demo:
51
 
52
  self.model_dropdown = gr.Dropdown(
53
  label="ESD Model",
54
- choices=['Van Gogh', 'Car'],
55
  value='Van Gogh',
56
  interactive=True
57
  )
 
51
 
52
  self.model_dropdown = gr.Dropdown(
53
  label="ESD Model",
54
+ choices= list(model_map.keys()),
55
  value='Van Gogh',
56
  interactive=True
57
  )