Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def get_emotion(microphone, file_upload, task):
|
|
20 |
|
21 |
file = microphone if microphone is not None else file_upload
|
22 |
print(file)
|
23 |
-
test = feature_extractor(file, sampling_rate=16000, padding=True, return_tensors="pt" )
|
24 |
logits = model(**test).logits
|
25 |
predicted_ids = torch.argmax(logits, dim=-1)
|
26 |
labels = [model.config.id2label[_id] for _id in predicated_ids.tolist()]
|
|
|
20 |
|
21 |
file = microphone if microphone is not None else file_upload
|
22 |
print(file)
|
23 |
+
test = feature_extractor(file, sampling_rate=16000, padding=True, return_tensors="pt" )
|
24 |
logits = model(**test).logits
|
25 |
predicted_ids = torch.argmax(logits, dim=-1)
|
26 |
labels = [model.config.id2label[_id] for _id in predicated_ids.tolist()]
|