Update app.py

app.py

@@ -43,7 +43,7 @@ def denoise_speech(audio):
             torch.nn.functional.pad(chunk, (40, 40)), return_tensors="pt"
         ).to('cuda')
         with torch.inference_mode():
-            feature = fe(inputs["input_features"])["last_hidden_state"]
+            feature = fe(inputs["input_features"].to("cuda"))["last_hidden_state"]
             if feature_cache is not None:
                 feature = torch.cat([feature_cache, feature], dim=1)
                 restored_wav = decoder(feature.transpose(1, 2))