Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def Page():
|
|
43 |
solara.Markdown(f'{spans1}')
|
44 |
outputs = model.generate(tokens, max_new_tokens=1, output_scores=True, return_dict_in_generate=True, pad_token_id=tokenizer.eos_token_id)
|
45 |
scores = F.softmax(outputs.scores[0], dim=-1)
|
46 |
-
top_10 = torch.topk(scores, 10)+torch.
|
47 |
df = pd.DataFrame()
|
48 |
df["probs"] = top_10.values[0]
|
49 |
df["probs"] = [f"{value:.2%}" for value in df["probs"].values]
|
|
|
43 |
solara.Markdown(f'{spans1}')
|
44 |
outputs = model.generate(tokens, max_new_tokens=1, output_scores=True, return_dict_in_generate=True, pad_token_id=tokenizer.eos_token_id)
|
45 |
scores = F.softmax(outputs.scores[0], dim=-1)
|
46 |
+
top_10 = torch.topk(scores, 10)+torch.topk(scores, 10, largest=False)
|
47 |
df = pd.DataFrame()
|
48 |
df["probs"] = top_10.values[0]
|
49 |
df["probs"] = [f"{value:.2%}" for value in df["probs"].values]
|