stats-dev commited on
Commit
c9fd2e0
1 Parent(s): 506bd46

test for requirements.txt

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # Load model directly
2
  import gradio as gr
3
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("j5ng/kcbert-formal-classifier")
6
  model = AutoModelForSequenceClassification.from_pretrained("j5ng/kcbert-formal-classifier")
 
1
  # Load model directly
2
  import gradio as gr
3
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("j5ng/kcbert-formal-classifier")
6
  model = AutoModelForSequenceClassification.from_pretrained("j5ng/kcbert-formal-classifier")