Nymbo commited on
Commit
87c83eb
1 Parent(s): 81e1739

adding OpenDalle, removing OpenJourney and Arcane

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -12,7 +12,7 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
12
  API_TOKEN = os.getenv("HF_READ_TOKEN")
13
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
14
  timeout = 100
15
- models_list = ["DALL-E 3 XL", "New Reality XL NSFW", "Juggernaut XL", "SDXL 1.0", "AbsoluteReality 1.8.1", "SSD-1B", "Dreamshaper XL Turbo", "Edge of Realism", "Playground 2", "NSFW Hentai", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "Disney", "CleanLinearMix", "Redmond SDXL", "Arcane"]
16
 
17
  # OP's other AI projects: https://ai-hub.rf.gd .
18
 
@@ -78,14 +78,14 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
78
  print(f'\033[1mГенерация {key}:\033[0m {prompt}')
79
  if model == 'DALL-E 3 XL':
80
  API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 
 
81
  if model == 'Playground 2':
82
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
83
  if model == 'Dreamshaper XL Turbo':
84
  API_URL = "https://api-inference.huggingface.co/models/Lykon/dreamshaper-xl-turbo"
85
  if model == 'SSD-1B':
86
  API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
87
- if model == 'Openjourney 4':
88
- API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
89
  if model == 'AbsoluteReality 1.8.1':
90
  API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
91
  if model == 'Lyriel 1.6':
@@ -119,8 +119,6 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
119
  API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
120
  if model == 'Redmond SDXL':
121
  API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
122
- if model == 'Arcane':
123
- API_URL = "https://api-inference.huggingface.co/models/nitrosocke/Arcane-Diffusion"
124
  if model == 'SDXL 1.0':
125
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/stable-diffusion-xl-base-1.0"
126
  if model == 'Edge of Realism':
 
12
  API_TOKEN = os.getenv("HF_READ_TOKEN")
13
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
14
  timeout = 100
15
+ models_list = ["DALL-E 3 XL", "OpenDalle", "New Reality XL NSFW", "Juggernaut XL", "SDXL 1.0", "AbsoluteReality 1.8.1", "SSD-1B", "Dreamshaper XL Turbo", "Edge of Realism", "Playground 2", "NSFW Hentai", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "Disney", "CleanLinearMix", "Redmond SDXL"]
16
 
17
  # OP's other AI projects: https://ai-hub.rf.gd .
18
 
 
78
  print(f'\033[1mГенерация {key}:\033[0m {prompt}')
79
  if model == 'DALL-E 3 XL':
80
  API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
81
+ if model == 'OpenDalle':
82
+ API_URL = "https://api-inference.huggingface.co/models/dataautogpt3/OpenDalle"
83
  if model == 'Playground 2':
84
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
85
  if model == 'Dreamshaper XL Turbo':
86
  API_URL = "https://api-inference.huggingface.co/models/Lykon/dreamshaper-xl-turbo"
87
  if model == 'SSD-1B':
88
  API_URL = "https://api-inference.huggingface.co/models/segmind/SSD-1B"
 
 
89
  if model == 'AbsoluteReality 1.8.1':
90
  API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
91
  if model == 'Lyriel 1.6':
 
119
  API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
120
  if model == 'Redmond SDXL':
121
  API_URL = "https://api-inference.huggingface.co/models/artificialguybr/LogoRedmond-LogoLoraForSDXL-V2"
 
 
122
  if model == 'SDXL 1.0':
123
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/stable-diffusion-xl-base-1.0"
124
  if model == 'Edge of Realism':