Hugging Face's logo

Spaces:
Mr-Vicky-01
/
AI_Artist
Running

App Files Community
1
Mr-Vicky-01 commited on
Commit
0ae1bd2
·
verified ·
1 Parent(s): 62bb1f8

Update image_engine.py

Browse files
Files changed (1) hide show
  1. image_engine.py +2 -1
image_engine.py CHANGED
@@ -6,7 +6,8 @@ import os
6
  # Define the ImageGenerator class for Hugging Face API
7
  class ImageGenerator:
8
  def __init__(self) -> None:
9
- self.api_url = "https://api-inference.huggingface.co/models/sd-community/sdxl-flash"
 
10
  self.headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
11
 
12
  def generate_image(self, prompt):
 
6
  # Define the ImageGenerator class for Hugging Face API
7
  class ImageGenerator:
8
  def __init__(self) -> None:
9
+ self.ape_url = "https://api-inference.huggingface.co/models/stablediffusionapi/mklan-xxx-nsfw-pony"
10
+ # self.api_url = "https://api-inference.huggingface.co/models/sd-community/sdxl-flash"
11
  self.headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
12
 
13
  def generate_image(self, prompt):