Kseniia-Kholina's picture
Create app.py
be331b6 verified
raw
history blame
495 Bytes
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()