timestamps

main.py

@@ -34,12 +34,12 @@ async def download_video(video_url: str):
     download_convert_video_to_audio(yt_dlp, video_url, f"/home/user/{uuid.uuid4().hex}")
 
 @app.post("/transcribe")
-async def transcribe_video(video_url: str, beam_size: int = 5):
+async def transcribe_video(video_url: str, beam_size: int = 5, word_timestamps: bool = True):
     print("doing hex")
     rand_id = uuid.uuid4().hex
     print("doing download")
     download_convert_video_to_audio(yt_dlp, video_url, f"/home/user/{rand_id}")
-    segments, info = model.transcribe(f"/home/user/{rand_id}.mp3", beam_size=beam_size, word_timestamps=True)
+    segments, info = model.transcribe(f"/home/user/{rand_id}.mp3", beam_size=beam_size, word_timestamps=word_timestamps)
     segments = [segment_to_dict(segment) for segment in segments]
     total_duration = round(info.duration, 2)  # Same precision as the Whisper timestamps.
     print(info)