matthewkram commited on
Commit
626743e
·
verified ·
1 Parent(s): 1a20d1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
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 (замена enable_vae_slicing)
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("""