Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from streamlit_player import st_player
 from transformers import pipeline
-
+from IPython.display import YouTubeVideo
 
 
 def tester(text):
@@ -27,7 +27,8 @@ def tester(text):
     st_player("https://www.youtube.com/watch?v=1k8craCGpgs")
     
   elif (results[0]['label']=="sadness"):
-    st.write(<iframe width="560" height="315" src="https://www.youtube.com/embed/kkIabnfvJI8" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>)
+    video = YouTubeVideo("1k8craCGpgs")
+    display(video)
     #st_player("https://www.youtube.com/embed/BZsXcc_tC-o?autoplay=1")
     
   elif (results[0]['label']=="surprise"):