Spaces:
Sleeping
Sleeping
Commit
•
99ea4aa
1
Parent(s):
f41562b
Update app.py
Browse files
app.py
CHANGED
@@ -117,7 +117,7 @@ def predict(one_text ):
|
|
117 |
|
118 |
# 使用模型进行预测
|
119 |
with torch.no_grad():
|
120 |
-
output =
|
121 |
# print(output)
|
122 |
# 计算预测概率
|
123 |
pred_score = nn.functional.softmax(output[0], dim=0)
|
|
|
117 |
|
118 |
# 使用模型进行预测
|
119 |
with torch.no_grad():
|
120 |
+
output = model_big_load(one_ids)
|
121 |
# print(output)
|
122 |
# 计算预测概率
|
123 |
pred_score = nn.functional.softmax(output[0], dim=0)
|