Update app.py

app.py

@@ -17,7 +17,6 @@ from utils.unet import UNet3DConditionModel
 from utils.pipeline_magictime import MagicTimePipeline
 from utils.util import save_videos_grid, convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint, load_diffusers_lora_unet, convert_ldm_clip_text_model
 
-
 from huggingface_hub import snapshot_download
 
 model_path = "ckpts"
@@ -77,7 +76,8 @@ os.system(f"rm -rf gradio_cached_examples/")
 device = "cuda"
 
 def random_seed():
-    return random.randint(1, 10**16)
+    # 修复：转成字符串以匹配 Textbox 组件的预期输入类型
+    return str(random.randint(1, 10**16))
 
 class MagicTimeController:
     def __init__(self):
@@ -152,15 +152,12 @@ class MagicTimeController:
         self.unet = Swift.from_pretrained(self.unet, magic_adapter_t_path)
         self.text_encoder = Swift.from_pretrained(self.text_encoder, magic_text_encoder_path)
 
-        return gr.Dropdown()
-
     def update_motion_module(self, motion_module_dropdown):
         self.selected_motion_module = motion_module_dropdown
         motion_module_dropdown = os.path.join(self.motion_module_dir, motion_module_dropdown)
         motion_module_state_dict = torch.load(motion_module_dropdown, map_location="cpu")
         _, unexpected = self.unet.load_state_dict(motion_module_state_dict, strict=False)
         assert len(unexpected) == 0
-        return gr.Dropdown()
     
     def update_motion_module_2(self, motion_module_dropdown):
         self.selected_motion_module = motion_module_dropdown
@@ -168,7 +165,6 @@ class MagicTimeController:
         motion_module_state_dict = torch.load(motion_module_dropdown, map_location="cpu")
         _, unexpected = self.unet_model.load_state_dict(motion_module_state_dict, strict=False)
         assert len(unexpected) == 0
-        return gr.Dropdown()
 
     @spaces.GPU(duration=300)
     def magictime(
@@ -199,8 +195,8 @@ class MagicTimeController:
         ).to(device)     
 
         if int(seed_textbox) > 0: seed = int(seed_textbox)
-        else: seed = random_seed()
-        torch.manual_seed(int(seed))
+        else: seed = int(random_seed())
+        torch.manual_seed(seed)
         
         assert seed == torch.initial_seed()
         print(f"### seed: {seed}")
@@ -233,7 +229,9 @@ class MagicTimeController:
 
         torch.cuda.empty_cache()
         time.sleep(1)
-        return gr.Video(value=save_sample_path), gr.Json(value=json_config)
+        
+        # 修复：直接返回文件路径和字典，不要返回包裹的 gr 组件对象
+        return save_sample_path, json_config
 
 controller = MagicTimeController()
 
@@ -292,4 +290,4 @@ def ui():
 if __name__ == "__main__":
     demo = ui()
     demo.queue(max_size=20)
-    demo.launch()
+    demo.launch(share=True) # 加上了 share=True 用于创建公开链接