saritha5 commited on
Commit
2bd7c5a
1 Parent(s): b6e152e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ def preprocess(text):
17
 
18
  # load model
19
  MODEL = f"ThirdEyeData/Complaints_Roberta"
20
- model = PegasusForConditionalGeneration.from_pretrained(MODEL)
21
- #model = AutoModelForSequenceClassification.from_pretrained(MODEL)
22
  #model.save_pretrained(MODEL)
23
 
24
 
 
17
 
18
  # load model
19
  MODEL = f"ThirdEyeData/Complaints_Roberta"
20
+ #model = PegasusForConditionalGeneration.from_pretrained(MODEL)
21
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL)
22
  #model.save_pretrained(MODEL)
23
 
24