JoshuaZywoo commited on
Commit
0443ecd
·
verified ·
1 Parent(s): da8d9a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import re
5
  # Load models
6
  emotion_classifier = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
7
  intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
8
- text_generator = pipeline("text2text-generation", model="google/flan-t5-base", max_new_tokens=150)
9
 
10
  # Candidate intents
11
  candidate_tasks = [
 
5
  # Load models
6
  emotion_classifier = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
7
  intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
8
+ text_generator = pipeline("text2text-generation", model="declare-lab/flan-alpaca-base", max_new_tokens=200)
9
 
10
  # Candidate intents
11
  candidate_tasks = [