ehdwns1516
commited on
Commit
·
41f1703
1
Parent(s):
f377110
Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,7 @@ def run_model(candicates_count, context: str, candicates: list[str]):
|
|
49 |
return {"result": candicates[torch.argmax(output.logits).item()]}
|
50 |
|
51 |
items = list()
|
52 |
-
count = 4
|
53 |
context = "your context"
|
54 |
for i in range(int(count)):
|
55 |
items.append("sentence")
|
|
|
49 |
return {"result": candicates[torch.argmax(output.logits).item()]}
|
50 |
|
51 |
items = list()
|
52 |
+
count = 4 # candicates count
|
53 |
context = "your context"
|
54 |
for i in range(int(count)):
|
55 |
items.append("sentence")
|