HH-ImgGen

Running on CPU Upgrade

Rooni commited on Dec 9, 2023

Commit

9e2ce11

•

1 Parent(s): 8f94034

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def query(prompt, model, is_negative=False, steps=1, cfg_scale=6, seed=None):
     if model == 'NewReality XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
     if model == 'Anything 5.0':
-        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/anything-v5"
     payload = {
         "inputs": prompt,

     if model == 'NewReality XL':
         API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/newrealityxl-global-nsfw"
     if model == 'Anything 5.0':
+        API_URL = "https://api-inference.huggingface.co/models/hogiahien/anything-v5-edited"
     payload = {
         "inputs": prompt,