Emanai commited on
Commit
a6b9dcc
·
verified ·
1 Parent(s): e26f7c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import pipeline
3
  from transformers import AutoTokenizer
4
 
5
  # Load the fine-tuned model from Hugging Face Hub
6
- classifier = pipeline("text-classification", model="Emanai/my-finetuned-bert")
7
- tokenizer = AutoTokenizer.from_pretrained('Emanai/my-finetuned-bert')
8
 
9
  def classify_text(text):
10
  return classifier(text)
 
3
  from transformers import AutoTokenizer
4
 
5
  # Load the fine-tuned model from Hugging Face Hub
6
+ classifier = pipeline("text-classification", model="Emanai/my-finetuned-bert2")
7
+ tokenizer = AutoTokenizer.from_pretrained('Emanai/my-finetuned-bert2')
8
 
9
  def classify_text(text):
10
  return classifier(text)