Update app.py

app.py

@@ -117,7 +117,7 @@ def predict(one_text ):
 
   # 使用模型进行预测
   with torch.no_grad():
-      output = model_load(one_ids)
+      output = model_big_load(one_ids)
   # print(output)
   # 计算预测概率
   pred_score = nn.functional.softmax(output[0], dim=0)