mariagrandury commited on
Commit
fa5172e
1 Parent(s): 832ee1c

Fix bug in app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ st.markdown(
66
  """
67
  )
68
 
69
- model_name = st.selectbox("Model",MODELS.keys())
70
  model_url = MODELS[model_name]["url"]
71
 
72
  prompt = st.selectbox("Prompt", ["Random", "Custom"])
 
66
  """
67
  )
68
 
69
+ model_name = st.selectbox("Model", list(MODELS.keys()))
70
  model_url = MODELS[model_name]["url"]
71
 
72
  prompt = st.selectbox("Prompt", ["Random", "Custom"])