Update README.md

README.md

@@ -54,7 +54,6 @@ language:
             truncation=True
         )
         
-        # ... 其他代码不变
         
         # 取出输入对应的编码
         input_ids = encoding['input_ids'].to(device)
@@ -62,8 +61,8 @@ language:
         
         # 不计算梯度
         with torch.no_grad():
-            # 产生情感预测的logits
-            outputs = model(input_ids=input_ids, attention_mask=attention_mask)
+        # 产生情感预测的logits
+        outputs = model(input_ids=input_ids, attention_mask=attention_mask)
         
         # 使用softmax将logits转换为概率
         probs = softmax(outputs.logits, dim=1)