Geek7 commited on
Commit
e7f6911
1 Parent(s): ea81549

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,7 +4,10 @@ import os
4
  os.environ['HF_HOME'] = '/blabla/cache/'
5
 
6
  # Load the diffusion model
7
- pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
 
 
 
8
 
9
  def generate_image(prompt):
10
  # Generate the image based on the prompt
 
4
  os.environ['HF_HOME'] = '/blabla/cache/'
5
 
6
  # Load the diffusion model
7
+ pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
8
+
9
+
10
+
11
 
12
  def generate_image(prompt):
13
  # Generate the image based on the prompt