milyiyo commited on
Commit
63989d3
·
1 Parent(s): 3a65207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
- from diffusers import DiffusionPipeline
 
3
  import torch
4
  import numpy as np
5
  import PIL
@@ -25,7 +26,7 @@ import cv2
25
 
26
  print('\nDEBUG: Version: 1')
27
 
28
- pipeline = DiffusionPipeline.from_pretrained("fusing/latent-diffusion-text2im-large")
29
  generator = torch.manual_seed(42)
30
 
31
  FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=milyiyo.testing-diffusers" />'
 
1
  import gradio as gr
2
+ #from diffusers import DiffusionPipeline
3
+ from diffusers import LDMTextToImagePipeline
4
  import torch
5
  import numpy as np
6
  import PIL
 
26
 
27
  print('\nDEBUG: Version: 1')
28
 
29
+ pipeline = LDMTextToImagePipeline.from_pretrained("fusing/latent-diffusion-text2im-large")
30
  generator = torch.manual_seed(42)
31
 
32
  FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=milyiyo.testing-diffusers" />'