move to gpu in load_model

app.py

@@ -120,6 +120,12 @@ class FoleyController:
 
         self.pipeline.load_ip_adapter(fc_ckpt, subfolder='semantic', weight_name='semantic_adapter.bin', image_encoder_folder=None)
 
+        # move to gpu
+        self.time_detector.to(self.device)
+        self.pipeline.to(self.device)
+        self.vocoder.to(self.device)
+        self.image_encoder.to(self.device)
+
         gr.Info("Load Finish!")
         print("Load Finish!")
         self.loaded = True
@@ -138,13 +144,7 @@ class FoleyController:
         sample_step_slider,
         cfg_scale_slider,
         seed_textbox, 
-    ):
-        # move to gpu
-        self.time_detector.to(self.device)
-        self.pipeline.to(self.device)
-        self.vocoder.to(self.device)
-        self.image_encoder.to(self.device)
-
+    ): 
         vision_transform_list = [
             torchvision.transforms.Resize((128, 128)),
             torchvision.transforms.CenterCrop((112, 112)),