alexktrs commited on
Commit
4f2e372
·
1 Parent(s): 45e4e5f

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +30 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from diffusers import DDPMPipeline
3
+
4
+ model = 'alexktrs/CumulusCloudsGenerator'
5
+ device='cpu'
6
+
7
+ generator = DDPMPipeline.from_pretrained(model)
8
+ generator.to(device)
9
+
10
+ def generate(num_inference_steps):
11
+
12
+ image1 = generator(num_inference_steps=num_inference_steps).images[0]
13
+ image2 = generator(num_inference_steps=num_inference_steps).images[0]
14
+
15
+ return [image1, image2]
16
+
17
+ with gr.Blocks() as demo:
18
+ gr.Markdown("""
19
+
20
+ # Generate Cumulus Clouds
21
+
22
+ """)
23
+
24
+ gallery=gr.Gallery(type="pil")
25
+ slider=gr.Slider(label='Inference Steps', minimum=0, maximum=100)
26
+
27
+ btn = gr.Button("Generate Clouds")
28
+ btn.click(fn=generate, inputs=slider, outputs=gallery)
29
+
30
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ accelerate
2
+ diffusers
3
+ transformers