Spaces:

khadija3818
/

text-classification

Sleeping

khadija3818 commited on Dec 4, 2023

Commit

a84d362

•

1 Parent(s): 99454ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import pipeline
 def detect_emotion(sentence):
-    emotion_analyzer = pipeline("text-classification", model="mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-sentiment")
     result = emotion_analyzer(sentence)
     emotion = result[0]['label']
     return emotion

 from transformers import pipeline
 def detect_emotion(sentence):
+    emotion_analyzer = pipeline("text-classification")
     result = emotion_analyzer(sentence)
     emotion = result[0]['label']
     return emotion