chokiproai commited on
Commit
8d68db5
1 Parent(s): 07d31c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -12,7 +12,7 @@ list_models = [
12
  "Anything-V4",
13
  "Disney-Pixar-Cartoon",
14
  "Pixel-Art-XL",
15
- "dalle-3-xl",
16
  ]
17
 
18
  def generate_txt2img(current_model, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7,
@@ -22,6 +22,7 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
22
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
23
  elif current_model == "SDXL-1.0":
24
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
25
  elif current_model == "OpenJourney-V4":
26
  API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
27
  elif current_model == "Anything-V4":
@@ -210,7 +211,7 @@ with gr.Blocks(css=css) as demo:
210
  current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
211
 
212
  with gr.Row(elem_id="prompt-container"):
213
- text_prompt = gr.Textbox(label="Prompt", placeholder="a cute cat", lines=1, elem_id="prompt-text-input")
214
  text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
215
 
216
  with gr.Row():
 
12
  "Anything-V4",
13
  "Disney-Pixar-Cartoon",
14
  "Pixel-Art-XL",
15
+ "Dalle-3-XL",
16
  ]
17
 
18
  def generate_txt2img(current_model, prompt, is_negative=False, image_style="None style", steps=50, cfg_scale=7,
 
22
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
23
  elif current_model == "SDXL-1.0":
24
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
25
+ API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-refiner-1.0"
26
  elif current_model == "OpenJourney-V4":
27
  API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
28
  elif current_model == "Anything-V4":
 
211
  current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[1])
212
 
213
  with gr.Row(elem_id="prompt-container"):
214
+ text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", lines=1, elem_id="prompt-text-input")
215
  text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
216
 
217
  with gr.Row():