Update app.py

app.py

@@ -5,10 +5,15 @@ import pytube as pt
 pipe = pipeline(model="Hoft/whisper-small-swedish-asr")  # change to "your-username/the-name-you-picked"
 sa = pipeline('sentiment-analysis', model='marma/bert-base-swedish-cased-sentiment')
 
+def get_emoji(feeling):
+    if feeling == 'POSITIVE':
+        return '😊'
+    else:
+        return '😔'
 def microphone_or_file_transcribe(audio):
     text = pipe(audio)["text"]
     sa_result = sa(text)[0]
-    return text, sa_result['label']
+    return text, get_emoji(sa_result['label'])
     
 def youtube_transcribe(url):
     yt = pt.YouTube(url)
@@ -18,7 +23,8 @@ def youtube_transcribe(url):
 
     text = pipe("audio.mp3")["text"]
 
-    return text
+    sa_result = sa(text)[0]
+    return text, get_emoji(sa_result['label'])
 
 
 app = gr.Blocks()