clarkchan commited on
Commit
33011f2
·
1 Parent(s): 07cda19
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
- tokenizer = AutoTokenizer.from_pretrained("distilbert-base-finetuned-sst-2-english")
4
  def greet(name):
5
  result = tokenizer([name],padding=True,truncation=True,return_tensors="pt")
6
  return result.input_ids[0]
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer
3
+ tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
4
  def greet(name):
5
  result = tokenizer([name],padding=True,truncation=True,return_tensors="pt")
6
  return result.input_ids[0]