Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
import os
|
| 2 |
import sys
|
| 3 |
import uuid
|
| 4 |
import shutil
|
|
@@ -28,10 +28,11 @@ class WanAnimateApp:
|
|
| 28 |
return None, "Upload both image and video."
|
| 29 |
|
| 30 |
try:
|
| 31 |
-
# Обработка изображения
|
| 32 |
if isinstance(ref_img, Image.Image):
|
| 33 |
ref_image = ref_img.convert("RGB").resize((576, 320))
|
| 34 |
else:
|
|
|
|
| 35 |
ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))
|
| 36 |
|
| 37 |
# Извлечение motion из видео
|
|
@@ -56,7 +57,7 @@ class WanAnimateApp:
|
|
| 56 |
num_inference_steps=num_steps,
|
| 57 |
num_frames=num_frames,
|
| 58 |
generator=generator,
|
| 59 |
-
decode_chunk_size=2, # Оптимизация для VAE
|
| 60 |
noise_aug_strength=noise_aug_strength
|
| 61 |
).frames[0]
|
| 62 |
|
|
@@ -70,6 +71,9 @@ class WanAnimateApp:
|
|
| 70 |
return None, f"Failed: {str(e)}"
|
| 71 |
|
| 72 |
def start_app():
|
|
|
|
|
|
|
|
|
|
| 73 |
app = WanAnimateApp()
|
| 74 |
with gr.Blocks(title="Wan2.2-Animate (Local No API)") as demo:
|
| 75 |
gr.HTML("""
|
|
|
|
| 1 |
+
import os # Добавили для создания директории
|
| 2 |
import sys
|
| 3 |
import uuid
|
| 4 |
import shutil
|
|
|
|
| 28 |
return None, "Upload both image and video."
|
| 29 |
|
| 30 |
try:
|
| 31 |
+
# Обработка изображения (используем только PIL, так как type="pil")
|
| 32 |
if isinstance(ref_img, Image.Image):
|
| 33 |
ref_image = ref_img.convert("RGB").resize((576, 320))
|
| 34 |
else:
|
| 35 |
+
# Если не PIL (редко на HF), пробуем открыть как путь
|
| 36 |
ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))
|
| 37 |
|
| 38 |
# Извлечение motion из видео
|
|
|
|
| 57 |
num_inference_steps=num_steps,
|
| 58 |
num_frames=num_frames,
|
| 59 |
generator=generator,
|
| 60 |
+
decode_chunk_size=2, # Оптимизация для VAE
|
| 61 |
noise_aug_strength=noise_aug_strength
|
| 62 |
).frames[0]
|
| 63 |
|
|
|
|
| 71 |
return None, f"Failed: {str(e)}"
|
| 72 |
|
| 73 |
def start_app():
|
| 74 |
+
# Создаём директорию для Gradio временных файлов (фикс ошибки)
|
| 75 |
+
os.makedirs("/tmp/gradio", exist_ok=True)
|
| 76 |
+
|
| 77 |
app = WanAnimateApp()
|
| 78 |
with gr.Blocks(title="Wan2.2-Animate (Local No API)") as demo:
|
| 79 |
gr.HTML("""
|