Rooni commited on
Commit
3ab0852
1 Parent(s): d077173

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 == 'custom':
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 = {