youngtsai commited on
Commit
db681b1
1 Parent(s): 6a6dfe0

screenshot_path = f'{OUTPUT_PATH}/{youtube_id}_{snapshot_sec}.jpg'

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -134,16 +134,12 @@ def download_youtube_video(youtube_id, output_path=OUTPUT_PATH):
134
 
135
 
136
  def screenshot_youtube_video(youtube_id, snapshot_sec):
137
- # 由于在 Hugging Face Spaces 中直接下载 YouTube 视频可能不可行,
138
- # 您可能需要调整这部分代码,例如允许用户上传视频。
139
- # ... 代码 ...
140
-
141
  # 这里假设视频已经在适当的位置
142
  video_path = f'{OUTPUT_PATH}/{youtube_id}.mp4'
143
 
144
  # Load the video and take a screenshot
145
  with VideoFileClip(video_path) as video:
146
- screenshot_path = f'{OUTPUT_PATH}/screenshots/{youtube_id}_{snapshot_sec}.jpg'
147
  video.save_frame(screenshot_path, snapshot_sec)
148
 
149
  return screenshot_path
 
134
 
135
 
136
  def screenshot_youtube_video(youtube_id, snapshot_sec):
 
 
 
 
137
  # 这里假设视频已经在适当的位置
138
  video_path = f'{OUTPUT_PATH}/{youtube_id}.mp4'
139
 
140
  # Load the video and take a screenshot
141
  with VideoFileClip(video_path) as video:
142
+ screenshot_path = f'{OUTPUT_PATH}/{youtube_id}_{snapshot_sec}.jpg'
143
  video.save_frame(screenshot_path, snapshot_sec)
144
 
145
  return screenshot_path