Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def preprocess(text):
|
|
14 |
return " ".join(new_text)
|
15 |
|
16 |
# load model
|
17 |
-
MODEL = f"
|
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 |
|