Spaces:

LutherYTT
/

Cantonese-Sentiment-Analysis-System-Demo

Sleeping

LutherYTT commited on Jul 16

Commit

b5ca05a

1 Parent(s): 980d5d0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ base_model = AutoModel.from_pretrained("hfl/chinese-roberta-wwm-ext")
 model = MultiTaskRoberta(base_model)
 # 加载权重
-model_path = "/content/robert-distilled-model/model.safetensors"
 state_dict = load_file(model_path, device="cpu")
 model.load_state_dict(state_dict)
 model.to(device)

 model = MultiTaskRoberta(base_model)
 # 加载权重
+model_path = "model.safetensors"
 state_dict = load_file(model_path, device="cpu")
 model.load_state_dict(state_dict)
 model.to(device)