loge-dot commited on
Commit
06f70ac
·
1 Parent(s): fad352a

提交代码更改

Browse files
Files changed (1) hide show
  1. 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, audio_inputs) # (1, 7) logits
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))}