saritha5 commited on
Commit
59e7461
1 Parent(s): ad67ce6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def preprocess(text):
14
  return " ".join(new_text)
15
 
16
  # load model
17
- MODEL = f"cardiffnlp/twitter-roberta-base-sentiment-latest"
18
  model = AutoModelForSequenceClassification.from_pretrained(MODEL)
19
  #model.save_pretrained(MODEL)
20
 
 
14
  return " ".join(new_text)
15
 
16
  # load model
17
+ MODEL = f"ThirdEyeData/Consumer-Complaint-Segmentation"
18
  model = AutoModelForSequenceClassification.from_pretrained(MODEL)
19
  #model.save_pretrained(MODEL)
20