ankush-003 commited on
Commit
0749cb7
1 Parent(s): 99c7040
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import tensorflow as tf
4
  # from transformers import TFAutoModelForSequenceClassification
5
 
6
  # Load model directly
7
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
 
9
  # tokenizer = AutoTokenizer.from_pretrained("ankush-003/nosqli_identifier")
10
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
4
  # from transformers import TFAutoModelForSequenceClassification
5
 
6
  # Load model directly
7
+ from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
8
 
9
  # tokenizer = AutoTokenizer.from_pretrained("ankush-003/nosqli_identifier")
10
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")