Spaces:

JoshuaZywoo
/

5240FInalProject

Sleeping

JoshuaZywoo commited on Mar 27

Commit

4947cb4

verified ·

1 Parent(s): 1e89726

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,8 @@ emotion_classifier = pipeline(
     return_all_scores=True
 )
 intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
-text_generator = pipeline("text2text-generation", model="google/flan-t5-large")
 candidate_tasks = [
     "change mobile plan", "top up balance", "report service outage",

     return_all_scores=True
 )
 intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
 candidate_tasks = [
     "change mobile plan", "top up balance", "report service outage",