Update model.py

model.py

@@ -43,7 +43,7 @@ class Wav2VecModel(nn.Module):
         )
 
     def forward(self, x):
-        x = self.processor(x, sampling_rate=16000, return_tensors="pt")["input_values"].squeeze(0).to("cuda")
+        x = self.processor(x, sampling_rate=16000, return_tensors="pt")["input_values"].squeeze(0)
         x = self.encoder(x).last_hidden_state
         x = torch.mean(x, dim=1)
         logits = self.intent_classifier(x)
@@ -67,7 +67,7 @@ class HubertSSLModel(nn.Module):
         )
 
     def forward(self, x):
-        x = self.processor(x, sampling_rate=16000, return_tensors="pt")["input_values"].squeeze(0).to("cuda")
+        x = self.processor(x, sampling_rate=16000, return_tensors="pt")["input_values"].squeeze(0)
         x = self.encoder(x).last_hidden_state
         x = torch.mean(x, dim=1)
         logits = self.intent_classifier(x)