Hugging Face's logo

Spaces:
Mr-Vicky-01
/
AI_Artist
Running

App Files Community
Mr-Vicky-01 commited on
Commit
80f487d
•
1 Parent(s): d7c8eb7

Update image_engine.py

Browse files
Files changed (1) hide show
  1. image_engine.py +2 -2
image_engine.py CHANGED
@@ -6,8 +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/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):
 
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/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):