Spaces:
Paused
Paused
123
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from transformers import AutoTokenizer
|
| 3 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 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]
|