Upload README.md

README.md

@@ -105,8 +105,7 @@ text = "孔子\n大成至圣先师孔丘，字仲尼，子姓，孔氏，敬称
 
 output = llm.classify(text)[0]
 probs = output.outputs.probs
-logits = torch.tensor(output.outputs.probs)
-probabilities = F.softmax(logits, dim=0)
+probabilities = torch.tensor(output.outputs.probs)
 
 # Get the top predicted class
 top_idx = torch.argmax(probabilities).item()