Kvikontent commited on
Commit
ac65298
·
1 Parent(s): 2f92c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from diffusers import DiffusionPipeline
3
 
4
  # Initialize the DiffusionPipeline with the specified model
5
- pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/juggernaut-xl-v5")
6
- pipeline.load_lora_weights("openskyml/dalle-3-xl")
7
 
8
  # Function to generate image based on the prompt
9
  def generate_image(prompt):
 
2
  from diffusers import DiffusionPipeline
3
 
4
  # Initialize the DiffusionPipeline with the specified model
5
+ pipeline = DiffusionPipeline.from_pretrained("Lykon/dreamshaper-7")
6
+ pipeline.load_lora_weights("openskyml/midjourney-v4-xl")
7
 
8
  # Function to generate image based on the prompt
9
  def generate_image(prompt):