Update main.py

main.py

@@ -21,19 +21,19 @@ def segment_to_dict(segment):
 
 @app.post("/video")
 async def download_video(video_url: str):
-    download_convert_video_to_audio(yt_dlp, video_url, f"/workspace/convo/videos/{uuid.uuid4().hex}")
+    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):
     print("doing hex")
     rand_id = uuid.uuid4().hex
     print("doing download")
-    download_convert_video_to_audio(yt_dlp, video_url, f"/workspace/convo/videos/{rand_id}")
-    segments, info = model.transcribe(f"/workspace/convo/videos/{rand_id}.mp3", beam_size=beam_size, word_timestamps=True)
+    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 = [segment_to_dict(segment) for segment in segments]
     total_duration = round(info.duration, 2)  # Same precision as the Whisper timestamps.
     print(info)
-    os.remove(f"/workspace/convo/videos/{rand_id}.mp3")
+    os.remove(f"/home/user/{rand_id}.mp3")
     print("Detected language '%s' with probability %f" % (info.language, info.language_probability))
     
     return segments