Spaces:

SamanthaStorm
/

Tether

Running on Zero

SamanthaStorm commited on May 21

Commit

916e8c1

verified ·

1 Parent(s): 31de5e2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def get_emotion_profile(text):
 emotion_pipeline = hf_pipeline(
     "text-classification",
     model="j-hartmann/emotion-english-distilroberta-base",
-    top_k=None,
     truncation=True
 )

 emotion_pipeline = hf_pipeline(
     "text-classification",
     model="j-hartmann/emotion-english-distilroberta-base",
+    top_k=6,
     truncation=True
 )