JingyeChen commited on
Commit
966167e
1 Parent(s): 08f7d1b
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,10 +5,11 @@ import torch
5
  import gradio as gr
6
  import time
7
  from transformers import CLIPTextModel, CLIPTokenizer
8
- from diffusers import AutoencoderKL, DDPMScheduler, StableDiffusionPipeline, UNet2DConditionModel
9
  from tqdm import tqdm
10
  from PIL import Image
11
  from PIL import Image, ImageDraw, ImageFont
 
12
 
13
  import string
14
  alphabet = string.digits + string.ascii_lowercase + string.ascii_uppercase + string.punctuation + ' ' # len(aphabet) = 95
 
5
  import gradio as gr
6
  import time
7
  from transformers import CLIPTextModel, CLIPTokenizer
8
+ from diffusers import AutoencoderKL, DDPMScheduler, StableDiffusionPipeline, UNet2DConditionModel, DiffusionPipeline, LCMScheduler
9
  from tqdm import tqdm
10
  from PIL import Image
11
  from PIL import Image, ImageDraw, ImageFont
12
+ import random
13
 
14
  import string
15
  alphabet = string.digits + string.ascii_lowercase + string.ascii_uppercase + string.punctuation + ' ' # len(aphabet) = 95