File size: 495 Bytes
be331b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gradio as gr
def topn_tokens(sequence, n):
chars = []
for char in sequence:
chars.append(char)
return chars
demo = gr.Interface(
topn_tokens,
[
"text",
gr.Dropdown(["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12", "13", "14", "15", "16", "17", "18", "19", "20"]),
],
"dataframe",
description="Choose a number between 1-20 to display n tokens",
)
if __name__ == "__main__":
demo.launch() |