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