Nova-school commited on
Commit
2c7aa47
·
verified ·
1 Parent(s): ce8de4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.bottomk(scores, 10)
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]