Update app.py

app.py

@@ -14,21 +14,18 @@ os.environ["VECLIB_MAXIMUM_THREADS"] = "2"
 os.environ["NUMEXPR_NUM_THREADS"] = "2"
 
 # ==============================================================================
-# КРИТИЧНИЙ ФІКС 2 (ПРАВИЛЬНИЙ MONKEYPATCH): 
-# Створюємо кастомний КЛАС замість функції, щоб зберегти можливість наслідування для insightface.
+# КРИТИЧНИЙ ФІКС 2: Патч для 5D GridSample на CUDA.
 # ==============================================================================
 import onnxruntime as ort
 _orig_InferenceSession = ort.InferenceSession
 
 class PatchedInferenceSession(_orig_InferenceSession):
     def __init__(self, path_or_bytes, *args, **kwargs):
-        # Якщо ініціалізується модель деформації обличчя — примусово перемикаємо на CPU
         if isinstance(path_or_bytes, str) and "warping_spade" in path_or_bytes:
             print(f"🎯 [MONKEYPATCH] Forcing {path_or_bytes} to run strictly on CPUExecutionProvider!")
             kwargs["providers"] = ["CPUExecutionProvider"]
         super().__init__(path_or_bytes, *args, **kwargs)
 
-# Перезаписуємо класи в просторах імен ONNX Runtime
 ort.InferenceSession = PatchedInferenceSession
 if hasattr(ort, 'capi') and hasattr(ort.capi, 'onnxruntime_inference_collection'):
     ort.capi.onnxruntime_inference_collection.InferenceSession = PatchedInferenceSession
@@ -36,6 +33,19 @@ if hasattr(ort, 'capi') and hasattr(ort.capi, 'onnxruntime_inference_collection'
 
 import pdb
 import gradio as gr
+
+# ==============================================================================
+# КРИТИЧНИЙ ФІКС 3 (MONKEYPATCH FOR GR.INFO):
+# Вирішуємо конфлікт версій Gradio. Вирізаємо 'duration', якого немає в Gradio 4.36.1,
+# щоб уникнути TypeError на самому фініші генерації відео.
+# ==============================================================================
+_orig_Info = gr.Info
+def patched_Info(message, *args, **kwargs):
+    kwargs.pop('duration', None) # Видаляємо duration, якщо він переданий автором
+    return _orig_Info(message, *args, **kwargs)
+gr.Info = patched_Info
+# ==============================================================================
+
 import os.path as osp
 from omegaconf import OmegaConf