Rooni commited on
Commit
3e0bf53
1 Parent(s): 3341831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,6 +11,8 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
11
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
12
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
13
 
 
 
14
  def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
15
  language = detect(prompt)
16
 
@@ -40,9 +42,7 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
40
  API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
41
  if model == 'epiCRealism':
42
  API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
43
- if model == 'Aurora':
44
- API_URL = "https://api-inference.huggingface.co/models/SweetLuna/Aurora"
45
-
46
  payload = {
47
  "inputs": prompt,
48
  "is_negative": is_negative,
@@ -65,7 +65,7 @@ with gr.Blocks(css=css) as dalle:
65
  with gr.Row():
66
  with gr.Column(elem_id="prompt-container"):
67
  text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
68
- model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=["AbsoluteReality v1.8.1", "DALL-E 3 XL", "Playground v2", "Openjourney v4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit v2.5", "Realistic Vision 5.1", "Incursios v1.6", "Anime Detailer XL LoRA", "Aurora", "epiCRealism"])
69
 
70
 
71
  with gr.Tab("Расширенные настройки"):
 
11
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
12
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
13
 
14
+ models_list = ["AbsoluteReality v1.8.1", "DALL-E 3 XL", "Playground v2", "Openjourney v4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit v2.5", "Realistic Vision 5.1", "Incursios v1.6", "Anime Detailer XL LoRA", "epiCRealism"]
15
+
16
  def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
17
  language = detect(prompt)
18
 
 
42
  API_URL = "https://api-inference.huggingface.co/models/Linaqruf/anime-detailer-xl-lora"
43
  if model == 'epiCRealism':
44
  API_URL = "https://api-inference.huggingface.co/models/emilianJR/epiCRealism"
45
+
 
 
46
  payload = {
47
  "inputs": prompt,
48
  "is_negative": is_negative,
 
65
  with gr.Row():
66
  with gr.Column(elem_id="prompt-container"):
67
  text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
68
+ model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=models_list)
69
 
70
 
71
  with gr.Tab("Расширенные настройки"):