Claude commited on
Commit
6059b17
·
unverified ·
1 Parent(s): 6c4d34d

perf: Run on CPU to avoid ZeroGPU duration limits

Browse files
Files changed (1) hide show
  1. app.py +10 -19
app.py CHANGED
@@ -85,7 +85,8 @@ collection = _default_client.get_or_create_collection(
85
 
86
 
87
  def get_device():
88
- return "cuda" if torch.cuda.is_available() else "cpu"
 
89
 
90
 
91
  def get_whisper_model():
@@ -524,24 +525,14 @@ def _process_youtube_impl(
524
  return f"Error analyzing video: {error_msg}"
525
 
526
 
527
- # Apply ZeroGPU decorator if available
528
- if ZEROGPU_AVAILABLE:
529
- @spaces.GPU(duration=120) # 2 minute timeout for video processing
530
- def process_youtube(
531
- url: str,
532
- num_frames: int,
533
- session_state: SessionState | None = None,
534
- progress: gr.Progress = gr.Progress(),
535
- ) -> str:
536
- return _process_youtube_impl(url, num_frames, session_state, progress)
537
- else:
538
- def process_youtube(
539
- url: str,
540
- num_frames: int,
541
- session_state: SessionState | None = None,
542
- progress: gr.Progress = gr.Progress(),
543
- ) -> str:
544
- return _process_youtube_impl(url, num_frames, session_state, progress)
545
 
546
 
547
  def chat_with_videos(
 
85
 
86
 
87
  def get_device():
88
+ # Use CPU to avoid ZeroGPU duration limits
89
+ return "cpu"
90
 
91
 
92
  def get_whisper_model():
 
525
  return f"Error analyzing video: {error_msg}"
526
 
527
 
528
+ # Run on CPU to avoid ZeroGPU duration limits
529
+ def process_youtube(
530
+ url: str,
531
+ num_frames: int,
532
+ session_state: SessionState | None = None,
533
+ progress: gr.Progress = gr.Progress(),
534
+ ) -> str:
535
+ return _process_youtube_impl(url, num_frames, session_state, progress)
 
 
 
 
 
 
 
 
 
 
536
 
537
 
538
  def chat_with_videos(