Hugging Face's logo

Spaces:
Mr-Vicky-01
/
AI_Artist
Running

App Files Community
1
Mr-Vicky-01 commited on
Commit
63b3ced
·
verified ·
1 Parent(s): bd3234c

Update image_engine.py

Browse files
Files changed (1) hide show
  1. image_engine.py +1 -1
image_engine.py CHANGED
@@ -7,7 +7,7 @@ import os
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):
 
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/stabilityai/stable-diffusion-3.5-large"
11
  self.headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
12
 
13
  def generate_image(self, prompt):