saritha5 commited on
Commit
c759a2b
1 Parent(s): 546fac2

Update app.py

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