Ubuntu commited on
Commit
ebc5b64
·
1 Parent(s): 29c0f97

fix id for short

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -80,6 +80,8 @@ def embed_youtube(youtube_url):
80
  try:
81
  # video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
82
  video_id = youtube_url.split("v=")[1]
 
 
83
  print(video_id)
84
  embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
85
  return gr.update(value=embed_html, visible=True), "", None
 
80
  try:
81
  # video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
82
  video_id = youtube_url.split("v=")[1]
83
+ if 'short' in youtube_url:
84
+ video_id = video_id.split("/")[-1]
85
  print(video_id)
86
  embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
87
  return gr.update(value=embed_html, visible=True), "", None