Hugging Face's logo

Spaces:
RanM
/
text2image_1
Runtime error

App Files Community
RanM commited on
Commit
b4900fe
·
verified ·
1 Parent(s): bf5575b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from generate_prompts import generate_prompt
8
 
9
  # Load the model once at the start
10
  print("Loading the Stable Diffusion model...")
11
- model = StableDiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
12
  print("Model loaded successfully.")
13
 
14
  def generate_image(prompt, prompt_name):
 
8
 
9
  # Load the model once at the start
10
  print("Loading the Stable Diffusion model...")
11
+ model = StableDiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", safety_checker=None)
12
  print("Model loaded successfully.")
13
 
14
  def generate_image(prompt, prompt_name):