Thea231 commited on
Commit
bbfb109
·
verified ·
1 Parent(s): c023dec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import numpy as np
6
 
7
  # Load Hugging Face pipelines
8
  emotion_classifier = pipeline("text-classification", model="bhadresh-savani/bert-base-go-emotion")
9
- feedback_generator = pipeline("text2text-generation", model="mrm8488/t5-small-finetuned-emotion")
10
  text_to_audio = pipeline("text-to-speech", model="facebook/mms-tts-eng")
11
 
12
  # Streamlit app UI
 
6
 
7
  # Load Hugging Face pipelines
8
  emotion_classifier = pipeline("text-classification", model="bhadresh-savani/bert-base-go-emotion")
9
+ feedback_generator = pipeline("text2text-generation", model="facebook/blenderbot-3B")
10
  text_to_audio = pipeline("text-to-speech", model="facebook/mms-tts-eng")
11
 
12
  # Streamlit app UI