Spaces:
Running
Running
adding OpenDalle, removing OpenJourney and Arcane
Browse files
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", "
|
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':
|