majedk01 commited on
Commit
406ddd7
1 Parent(s): e38b507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -31,11 +31,13 @@ def generate_image(prompt, model_choice, translate=False):
31
  if translate:
32
  prompt = translate_text(prompt, target_language='en') # Assuming you want to translate to English
33
  model_urls = {
34
- "Stable Diffusion v1.5": "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
35
  "dalle-3-xl-v2": "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl-v2",
36
  "midjourney-v6": "https://api-inference.huggingface.co/models/Kvikontent/midjourney-v6",
37
  "openjourney-v4": "https://api-inference.huggingface.co/models/prompthero/openjourney-v4",
38
- "LCM_Dreamshaper_v7": "https://api-inference.huggingface.co/models/SimianLuo/LCM_Dreamshaper_v7"
 
 
39
 
40
  }
41
  API_URL = model_urls[model_choice]
@@ -76,7 +78,7 @@ iface = gr.Interface(
76
  inputs=[
77
  gr.components.Textbox(lines=2, placeholder="Enter the description of the image here..."),
78
  gr.components.Dropdown(choices=["Stable Diffusion v1.5","dalle-3-xl-v2","midjourney-v6","openjourney-v4","LCM_Dreamshaper_v7"], label="Choose Model", value='Stable Diffusion v1.5'),
79
- gr.components.Checkbox(label="Translate The Text Before Generating Image", value=False)
80
  ],
81
  outputs=gr.components.Image(),
82
  title=title,
 
31
  if translate:
32
  prompt = translate_text(prompt, target_language='en') # Assuming you want to translate to English
33
  model_urls = {
34
+ "Stable Diffusion v1.5": "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5",
35
  "dalle-3-xl-v2": "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl-v2",
36
  "midjourney-v6": "https://api-inference.huggingface.co/models/Kvikontent/midjourney-v6",
37
  "openjourney-v4": "https://api-inference.huggingface.co/models/prompthero/openjourney-v4",
38
+ "LCM_Dreamshaper_v7": "https://api-inference.huggingface.co/models/SimianLuo/LCM_Dreamshaper_v7",
39
+
40
+
41
 
42
  }
43
  API_URL = model_urls[model_choice]
 
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","dalle-3-xl-v2","midjourney-v6","openjourney-v4","LCM_Dreamshaper_v7"], label="Choose Model", value='Stable Diffusion v1.5'),
81
+
82
  ],
83
  outputs=gr.components.Image(),
84
  title=title,