Akshat7777 commited on
Commit
ac81f3a
·
verified ·
1 Parent(s): cd20c1e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from generate import generate
3
+
4
+ def infer(prompt, steps, guidance):
5
+ output = generate(prompt, steps, guidance)
6
+ return output
7
+
8
+ with gr.Blocks() as demo:
9
+ gr.Markdown("# 🌀 AnimateDiff Generator")
10
+
11
+ with gr.Row():
12
+ with gr.Column():
13
+ prompt = gr.Textbox(label="Prompt", placeholder="Describe your animation...")
14
+ steps = gr.Slider(10, 100, value=50, step=1, label="Inference Steps")
15
+ guidance = gr.Slider(1, 15, value=7.5, step=0.5, label="Guidance Scale")
16
+ btn = gr.Button("Generate")
17
+
18
+ with gr.Column():
19
+ output = gr.Image(label="Generated Frame")
20
+
21
+ btn.click(fn=infer, inputs=[prompt, steps, guidance], outputs=output)
22
+
23
+ if __name__ == "__main__":
24
+ demo.launch()
25
+