Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
# исправленная версия (чтобы не потерялась)
|
| 2 |
import subprocess
|
| 3 |
-
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 4 |
# subprocess.run('pip install bitsandbytes', shell=True)
|
| 5 |
-
subprocess.run('pip install av==12.0.0', shell=True)
|
| 6 |
|
| 7 |
import gradio as gr
|
| 8 |
import spaces
|
|
@@ -69,11 +69,11 @@ def sample(
|
|
| 69 |
base_count = len(glob(os.path.join(output_folder, "*.mp4")))
|
| 70 |
video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
|
| 71 |
res_variants = {
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
}
|
| 78 |
width = int(res_variants[resolution].split('x')[0])
|
| 79 |
height = int(res_variants[resolution].split('x')[1])
|
|
@@ -125,14 +125,6 @@ def resize_image(image, output_size=(672, 384)):
|
|
| 125 |
return cropped_image
|
| 126 |
|
| 127 |
with gr.Blocks() as demo:
|
| 128 |
-
aspect_ratio_variants = {
|
| 129 |
-
'16:9 (672x384)': '16:9',
|
| 130 |
-
'9:16 (384x672)': '9:16',
|
| 131 |
-
'1:1 (512x512)': '1:1',
|
| 132 |
-
'1:2 (352x736)': '1:2',
|
| 133 |
-
'2:1 (736x352)': '2:1'
|
| 134 |
-
}
|
| 135 |
-
|
| 136 |
gr.Markdown('''# Community demo for Kandinsky 4.0 Flash''')
|
| 137 |
with gr.Row():
|
| 138 |
with gr.Column():
|
|
@@ -157,7 +149,6 @@ with gr.Blocks() as demo:
|
|
| 157 |
choices=["16:9 (672x384)", "9:16 (384x672)", "1:1 (512x512)", "1:2 (352x736)", "2:1 (736x352)"],
|
| 158 |
value="16:9 (672x384)"
|
| 159 |
)
|
| 160 |
-
aspect_ratio = aspect_ratio_variants[aspect_ratio]
|
| 161 |
with gr.Column():
|
| 162 |
enhance_button = gr.Button("Beautify prompt with [GigaChat-Max](https://giga.chat)")
|
| 163 |
generate_btn = gr.Button("Generate Video")
|
|
|
|
| 1 |
# исправленная версия (чтобы не потерялась)
|
| 2 |
import subprocess
|
| 3 |
+
# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 4 |
# subprocess.run('pip install bitsandbytes', shell=True)
|
| 5 |
+
# subprocess.run('pip install av==12.0.0', shell=True)
|
| 6 |
|
| 7 |
import gradio as gr
|
| 8 |
import spaces
|
|
|
|
| 69 |
base_count = len(glob(os.path.join(output_folder, "*.mp4")))
|
| 70 |
video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
|
| 71 |
res_variants = {
|
| 72 |
+
'16:9 (672x384)': '672x384',
|
| 73 |
+
'9:16 (384x672)': '384x672',
|
| 74 |
+
'1:1 (512x512)': '512x512',
|
| 75 |
+
'1:2 (352x736)': '352x736',
|
| 76 |
+
'2:1 (736x352)': '736x352'
|
| 77 |
}
|
| 78 |
width = int(res_variants[resolution].split('x')[0])
|
| 79 |
height = int(res_variants[resolution].split('x')[1])
|
|
|
|
| 125 |
return cropped_image
|
| 126 |
|
| 127 |
with gr.Blocks() as demo:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
gr.Markdown('''# Community demo for Kandinsky 4.0 Flash''')
|
| 129 |
with gr.Row():
|
| 130 |
with gr.Column():
|
|
|
|
| 149 |
choices=["16:9 (672x384)", "9:16 (384x672)", "1:1 (512x512)", "1:2 (352x736)", "2:1 (736x352)"],
|
| 150 |
value="16:9 (672x384)"
|
| 151 |
)
|
|
|
|
| 152 |
with gr.Column():
|
| 153 |
enhance_button = gr.Button("Beautify prompt with [GigaChat-Max](https://giga.chat)")
|
| 154 |
generate_btn = gr.Button("Generate Video")
|