Hugging Face's logo

Spaces:
Mr-Vicky-01
/
AI_Artist
Running

App Files Community
1
Mr-Vicky-01 commited on
Commit
30d7589
·
verified ·
1 Parent(s): 44ad93d

Update image_engine.py

Browse files
Files changed (1) hide show
  1. image_engine.py +1 -0
image_engine.py CHANGED
@@ -9,6 +9,7 @@ class ImageGenerator:
9
  def __init__(self) -> None:
10
  self.all_urls = [
11
  "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell",
 
12
  "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
13
  ]
14
  self.api_url = random.choice(self.all_urls)
 
9
  def __init__(self) -> None:
10
  self.all_urls = [
11
  "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell",
12
+ "https://api-inference.huggingface.co/models/XLabs-AI/flux-RealismLora",
13
  "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
14
  ]
15
  self.api_url = random.choice(self.all_urls)