Update app.py

app.py

@@ -9,8 +9,6 @@ from huggingface_hub import hf_hub_download
 fe_path = hf_hub_download("sarulab-speech/sidon-v0.1", filename="feature_extractor.pt")
 decoder_path = hf_hub_download("sarulab-speech/sidon-v0.1", filename="decoder.pt")
 
-fe = torch.jit.load(fe_path,map_location='cuda').to('cuda')
-decoder = torch.jit.load(decoder_path,map_location='cuda').to('cuda')
 preprocessor = transformers.SeamlessM4TFeatureExtractor.from_pretrained(
     "facebook/w2v-bert-2.0"
 )
@@ -18,6 +16,9 @@ preprocessor = transformers.SeamlessM4TFeatureExtractor.from_pretrained(
 
 @spaces.GPU
 def denoise_speech(audio):
+    fe = torch.jit.load(fe_path,map_location='cuda').to('cuda')
+    decoder = torch.jit.load(decoder_path,map_location='cuda').to('cuda')
+
     if audio is None:
         return None