Link to model

gradio_server.py

@@ -56,7 +56,7 @@ def generate_video(
     samples = outputs['samples']
     sample = samples[0].unsqueeze(0)
     
-    save_path = os.path.join(os.getcwd(), "gradio_outputs")
+    save_path = "./gradio_outputs"
     os.makedirs(save_path, exist_ok=True)
     
     time_flag = datetime.fromtimestamp(time.time()).strftime("%Y-%m-%d-%H:%M:%S")
@@ -66,10 +66,18 @@ def generate_video(
     
     return video_path
 
-def create_demo(model_path, save_path):
+def create_demo(model_path):
     model = initialize_model(model_path)
     
     with gr.Blocks() as demo:
+        if torch.cuda.device_count() == 0:
+            with gr.Row():
+                gr.HTML("""
+                    <p style="background-color: red;"><big><big><big><b>⚠️To use <i>Hunyuan Video</i>, <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/HunyuanVideo?duplicate=true">duplicate this space</a> and set a GPU with 80 GB VRAM.</b>
+    
+                    You can't use <i>Hunyuan Video</i> directly here because this space runs on a CPU, which is not enough for <i>Hunyuan Video</i>. Please provide <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/HunyuanVideo/discussions/new">feedback</a> if you have issues.
+                    </big></big></big></p>
+                    """)
         gr.Markdown("# Hunyuan Video Generation")
         
         with gr.Row():
@@ -135,5 +143,5 @@ def create_demo(model_path, save_path):
 
 if __name__ == "__main__":
     os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-    demo = create_demo(args.model_base, args.save_path)
+    demo = create_demo("tencent/HunyuanVideo")
     demo.queue(10).launch()