duanyuxuan
commited on
Commit
•
811f398
1
Parent(s):
6774468
Fix env and format
Browse files- app.py +11 -2
- requirements.txt +1 -1
app.py
CHANGED
@@ -87,8 +87,17 @@ def preprocess_image(image, height = 512, width = 512):
|
|
87 |
image = image.resize((width, height))
|
88 |
return image
|
89 |
|
90 |
-
|
91 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
92 |
gr.Markdown(
|
93 |
"""
|
94 |
# Stable Video Diffusion distilled by ✨Target-Driven Distillation✨
|
|
|
87 |
image = image.resize((width, height))
|
88 |
return image
|
89 |
|
90 |
+
css = """
|
91 |
+
h1 {
|
92 |
+
text-align: center;
|
93 |
+
display:block;
|
94 |
+
}
|
95 |
+
.gradio-container {
|
96 |
+
max-width: 70.5rem !important;
|
97 |
+
}
|
98 |
+
"""
|
99 |
+
|
100 |
+
with gr.Blocks(css = css) as demo:
|
101 |
gr.Markdown(
|
102 |
"""
|
103 |
# Stable Video Diffusion distilled by ✨Target-Driven Distillation✨
|
requirements.txt
CHANGED
@@ -5,4 +5,4 @@ torchvision==0.16.0
|
|
5 |
transformers==4.42.3
|
6 |
xformers==0.0.22.post7
|
7 |
peft==0.9.0
|
8 |
-
imageio
|
|
|
5 |
transformers==4.42.3
|
6 |
xformers==0.0.22.post7
|
7 |
peft==0.9.0
|
8 |
+
imageio[ffmpeg]
|