ezb commited on
Commit
65d5285
1 Parent(s): 9a40e0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,9 +2,9 @@ from diffusers import StableDiffusionXLPipeline
2
  import torch
3
  import gradio as gr
4
 
5
- # Load the model
6
  model_id = "RunDiffusion/Juggernaut-X-v10"
7
- pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
8
  pipe = pipe.to("cpu")
9
 
10
  def text_to_image(prompt, negative_prompt, steps, guidance_scale, add_4k_masterpiece):
@@ -30,4 +30,5 @@ gradio_interface = gr.Interface(
30
  theme=gr.themes.Soft()
31
  )
32
 
33
- gradio_interface.launch(share=True, force_download=True)
 
 
2
  import torch
3
  import gradio as gr
4
 
5
+ # Load the model with forced download
6
  model_id = "RunDiffusion/Juggernaut-X-v10"
7
+ pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float32, force_download=True)
8
  pipe = pipe.to("cpu")
9
 
10
  def text_to_image(prompt, negative_prompt, steps, guidance_scale, add_4k_masterpiece):
 
30
  theme=gr.themes.Soft()
31
  )
32
 
33
+ # Launch the Gradio app with share=True
34
+ gradio_interface.launch(share=True)