fix(src): :rocket: Merge all into one function

app.py

@@ -9,8 +9,19 @@ from datetime import datetime
 import gradio as gr
 
 @spaces.GPU
-def initialize_model(model):
-    # quant_config = BitsAndBytesConfig(load_in_8bit=True)
+def generate_video(
+    prompt,
+    resolution,
+    video_length,
+    seed,
+    num_inference_steps,
+    guidance_scale,
+    flow_shift,
+    embedded_guidance_scale
+):
+    seed = None if seed == -1 else seed
+    width, height = resolution.split("x")
+    width, height = int(width), int(height)
 
     transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
         model,
@@ -28,24 +39,6 @@ def initialize_model(model):
         device_map="balanced",
     )
     
-    return pipeline
-
-@spaces.GPU
-def generate_video(
-    pipeline,
-    prompt,
-    resolution,
-    video_length,
-    seed,
-    num_inference_steps,
-    guidance_scale,
-    flow_shift,
-    embedded_guidance_scale
-):
-    seed = None if seed == -1 else seed
-    width, height = resolution.split("x")
-    width, height = int(width), int(height)
-    
     # Generar el video usando el pipeline
     video = pipeline(
         prompt=prompt,
@@ -70,8 +63,6 @@ def generate_video(
     return video_path
 
 def create_demo(model):
-    pipeline = initialize_model(model)
-    
     with gr.Blocks() as demo:
         gr.Markdown("# Hunyuan Video Generation")
         
@@ -120,7 +111,7 @@ def create_demo(model):
                 output = gr.Video(label="Generated Video")
         
         generate_btn.click(
-            fn=lambda *inputs: generate_video(pipeline, *inputs),
+            fn=lambda *inputs: generate_video(*inputs),
             inputs=[
                 prompt,
                 resolution,