tugot17 commited on
Commit
5cde343
1 Parent(s): 24866b4

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
 
4
  classifier = pipeline("text-classification", model="tugot17/my-awesome-model", tokenizer=tokenizer)
5
  def predict(text):
6
  output = {}
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
5
+
6
  classifier = pipeline("text-classification", model="tugot17/my-awesome-model", tokenizer=tokenizer)
7
  def predict(text):
8
  output = {}