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,
|