Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
|
|
64 |
API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
|
65 |
if model == 'CleanLinearMix':
|
66 |
API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
|
67 |
-
if model == '
|
68 |
API_URL == f"https://api-inference.huggingface.co/models/{custom}"
|
69 |
|
70 |
payload = {
|
|
|
64 |
API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
|
65 |
if model == 'CleanLinearMix':
|
66 |
API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
|
67 |
+
if model == 'Custom':
|
68 |
API_URL == f"https://api-inference.huggingface.co/models/{custom}"
|
69 |
|
70 |
payload = {
|