Update app.py

app.py

@@ -232,13 +232,15 @@ if args.remove_pretransform_weight_norm == "post_load":
 ckpt_path = hf_hub_download(repo_id="liuhuadai/ThinkSound", filename="thinksound.ckpt",repo_type="model")
 training_wrapper = create_training_wrapper_from_config(model_config, model)
 # 加载模型权重时根据设备选择map_location
-training_wrapper.load_state_dict(torch.load(ckpt_path)['state_dict']).to("cuda")
+training_wrapper.load_state_dict(torch.load(ckpt_path)['state_dict'])
+
+training_wrapper.to("cuda")
 
 def get_video_duration(video_path):
     video = VideoFileClip(video_path)
     return video.duration
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=60)
 @torch.inference_mode()
 @torch.no_grad()
 def get_audio(video_path, caption):