curtpond commited on
Commit
9fbed41
·
1 Parent(s): 2198b48

Debugging app.py code.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import pipeline
14
 
15
  # file name
16
  #lr_filename = 'logistic_regression.pkl'
17
- model = pipeline("sentiment-analysis", model="scikit-learn/skorch-text-classification")
18
 
19
  # Load model from pickle file
20
  # model = pickle.load(open(lr_filename, 'rb'))
 
14
 
15
  # file name
16
  #lr_filename = 'logistic_regression.pkl'
17
+ model = pipeline("sentiment-analysis", model="michellejieli/NSFW_text_classifier")
18
 
19
  # Load model from pickle file
20
  # model = pickle.load(open(lr_filename, 'rb'))