JoshuaZywoo commited on
Commit
a3d64e4
·
verified ·
1 Parent(s): dd50966

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ emotion_classifier = pipeline("text-classification", model="shengqizhao0124/emot
6
  intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
7
  text_generator = pipeline("text2text-generation", model="declare-lab/flan-alpaca-base")
8
 
 
9
  # Intent categories
10
  candidate_tasks = [
11
  "change mobile plan", "top up balance", "report service outage",
 
6
  intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
7
  text_generator = pipeline("text2text-generation", model="declare-lab/flan-alpaca-base")
8
 
9
+ pipeline("text-classification", model="shengqizhao0124/emotion_trainer", return_all_scores=True)
10
  # Intent categories
11
  candidate_tasks = [
12
  "change mobile plan", "top up balance", "report service outage",