Update app.py

app.py

@@ -16,7 +16,7 @@ SR = 16000
 
 
 def resample(speech_array, sampling_rate):
-    speech = torch.from_numpy(np.array([speech_array])).type(torch.ShortTensor)
+    speech = torch.from_numpy(np.array([speech_array])).type(torch.cuda.ShortTensor)
     print(speech, speech.shape, sampling_rate)
     resampler = torchaudio.transforms.Resample(sampling_rate)
     speech = resampler(speech).squeeze().numpy()
@@ -41,6 +41,7 @@ model = AutoModel.from_pretrained("Aniemore/wav2vec2-xlsr-53-russian-emotion-rec
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("Aniemore/wav2vec2-xlsr-53-russian-emotion-recognition")
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(device)
 
 
 def recognize(audio):