JoshuaZywoo commited on
Commit
365a2f1
·
verified ·
1 Parent(s): 0cfff50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load models
5
  emotion_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
6
  intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
7
- text_generator = pipeline("text2text-generation", model="google/flan-t5-large")
8
 
9
  # Intent categories
10
  candidate_tasks = [
 
4
  # Load models
5
  emotion_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
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 = [