CaioXapelaum commited on
Commit
6dc4c3e
·
verified ·
1 Parent(s): 69cbc25

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoTokenizer
3
+
4
+ # Function to load and tokenize the message using the selected tokenizer
5
+ def tokenize(message, history, tokenizer_name):
6
+ try:
7
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
8
+ return str(tokenizer.encode(message))
9
+ except Exception as e:
10
+ return f"Error loading tokenizer: {str(e)}"
11
+
12
+ # Gradio interface
13
+ def create_chat_interface():
14
+ with gr.Blocks() as demo:
15
+ tokenizer_input = gr.Textbox(placeholder="Enter the tokenizer name (e.g., gpt2, bert-base-uncased)", label="Tokenizer Name")
16
+ chatbot = gr.ChatInterface(fn=tokenize, inputs=["text", "state", tokenizer_input])
17
+ demo.append(tokenizer_input)
18
+ demo.append(chatbot)
19
+ return demo
20
+
21
+ # Launch the interface
22
+ if __name__ == "__main__":
23
+ demo = create_chat_interface()
24
+ demo.launch()