Hugging Face's logo

Spaces:
Gokce
/
Generated_text_Classifier_for_RestaurantReviewsTR
Build error

App Files Community
Gokce commited on
Commit
1990b99
·
1 Parent(s): af9f5a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
 
5
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("Gokce/finetuned-distilbert-multi-rr")
8
  model = AutoModelForSequenceClassification.from_pretrained("Gokce/finetuned-distilbert-multi-rr")
 
2
  from transformers import pipeline
3
 
4
 
5
+ from transformers import DistilBertTokenizer, TFAutoModelForSequenceClassification
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("Gokce/finetuned-distilbert-multi-rr")
8
  model = AutoModelForSequenceClassification.from_pretrained("Gokce/finetuned-distilbert-multi-rr")