nevreal commited on
Commit
719c01b
1 Parent(s): bcb28b5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def get_client(model_name):
9
 
10
  # Function to generate the image based on selected model and prompt
11
  def generate_image(prompt, model_name):
12
- client = get_client(model_options[model_name])
13
  response = client.text_to_image(prompt, guidance_scale=7.5)
14
  return response
15
 
 
9
 
10
  # Function to generate the image based on selected model and prompt
11
  def generate_image(prompt, model_name):
12
+ client = get_client(model_name)
13
  response = client.text_to_image(prompt, guidance_scale=7.5)
14
  return response
15