loge-dot
commited on
Commit
·
06f70ac
1
Parent(s):
fad352a
提交代码更改
Browse files- utils/model_inference.py +2 -1
utils/model_inference.py
CHANGED
@@ -38,6 +38,7 @@ model = MultimodalClassifier().to(device)
|
|
38 |
|
39 |
# 加载 SafeTensors 权重
|
40 |
state_dict = load_file(model_path)
|
|
|
41 |
model.load_state_dict(state_dict)
|
42 |
model.eval() # 设置为评估模式
|
43 |
|
@@ -58,7 +59,7 @@ def predict_emotion(text, audio):
|
|
58 |
audio_inputs = preprocess_audio(audio)
|
59 |
|
60 |
with torch.no_grad():
|
61 |
-
output = model(text_inputs
|
62 |
probabilities = F.softmax(output, dim=1).squeeze().tolist() # 归一化为概率
|
63 |
|
64 |
return {labels[i]: f"{probabilities[i]*100:.2f}%" for i in range(len(labels))}
|
|
|
38 |
|
39 |
# 加载 SafeTensors 权重
|
40 |
state_dict = load_file(model_path)
|
41 |
+
print("state_dict:", state_dict)
|
42 |
model.load_state_dict(state_dict)
|
43 |
model.eval() # 设置为评估模式
|
44 |
|
|
|
59 |
audio_inputs = preprocess_audio(audio)
|
60 |
|
61 |
with torch.no_grad():
|
62 |
+
output = model(audio_inputs,text_inputs) # (1, 7) logits
|
63 |
probabilities = F.softmax(output, dim=1).squeeze().tolist() # 归一化为概率
|
64 |
|
65 |
return {labels[i]: f"{probabilities[i]*100:.2f}%" for i in range(len(labels))}
|