clarkchan commited on
Commit
d3d9869
·
1 Parent(s): 33011f2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ 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]
7
 
8
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
9
  iface.launch()
 
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
+ tensor_str = ','.join(map(str, result.input_ids.numpy()))
7
 
8
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
9
  iface.launch()