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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import pipeline
3
 
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
 
 
2
  from transformers import pipeline
3
 
4
  # Load models
5
+ emotion_classifier = pipeline("text-classification", model="shengqizhao0124/emotion_trainer", 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