Mr-Vicky-01 commited on
Commit
d7c8eb7
1 Parent(s): 0ae1bd2

Update image_engine.py

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