majedk01 commited on
Commit
a1adc6b
1 Parent(s): 249e8d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,6 +36,7 @@ def generate_image(prompt, model_choice, translate=False):
36
  "sdxl-turbo": "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo",
37
  "MistoLine": "https://api-inference.huggingface.co/models/TheMistoAI/MistoLine",
38
  "Hyper-SD": "https://api-inference.huggingface.co/models/ByteDance/Hyper-SD",
 
39
 
40
 
41
  }
@@ -76,7 +77,7 @@ iface = gr.Interface(
76
  fn=generate_image,
77
  inputs=[
78
  gr.components.Textbox(lines=2, placeholder="Enter the description of the image here..."),
79
- gr.components.Dropdown(choices=["Stable Diffusion v1.5","stable-diffusion-xl-base-1.0","sdxl-turbo","MistoLine","Hyper-SD"], label="Choose Model", value='Stable Diffusion v1.5'),
80
  gr.components.Checkbox(label="Translate The Text Before Generating Image", value=False)
81
  ],
82
  outputs=gr.components.Image(),
 
36
  "sdxl-turbo": "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo",
37
  "MistoLine": "https://api-inference.huggingface.co/models/TheMistoAI/MistoLine",
38
  "Hyper-SD": "https://api-inference.huggingface.co/models/ByteDance/Hyper-SD",
39
+ "arabic-building": "https://api-inference.huggingface.co/models/jfachrel/arabic-building",
40
 
41
 
42
  }
 
77
  fn=generate_image,
78
  inputs=[
79
  gr.components.Textbox(lines=2, placeholder="Enter the description of the image here..."),
80
+ gr.components.Dropdown(choices=["Stable Diffusion v1.5","stable-diffusion-xl-base-1.0","sdxl-turbo","MistoLine","Hyper-SD","arabic-building"], label="Choose Model", value='Stable Diffusion v1.5'),
81
  gr.components.Checkbox(label="Translate The Text Before Generating Image", value=False)
82
  ],
83
  outputs=gr.components.Image(),