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