Rooni commited on
Commit
e3e5e4b
1 Parent(s): f00b283

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -11,7 +11,7 @@ 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
- models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL LoRA", "epiCRealism", "PixelArt XL", "NewReality XL", "Midjourney 4.0"]
15
 
16
  def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
17
  language = detect(prompt)
@@ -46,8 +46,6 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
46
  API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
47
  if model == 'NewReality XL':
48
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
49
- if model == 'Midjourney 4.0':
50
- API_URL = "https://api-inference.huggingface.co/models/openskyml/midjourney-v4-xl"
51
 
52
  payload = {
53
  "inputs": prompt,
 
11
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
12
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
13
 
14
+ models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL LoRA", "epiCRealism", "PixelArt XL", "NewReality XL"]
15
 
16
  def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
17
  language = detect(prompt)
 
46
  API_URL = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
47
  if model == 'NewReality XL':
48
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
 
 
49
 
50
  payload = {
51
  "inputs": prompt,