liuhuadai commited on
Commit
9ebfa72
·
verified ·
1 Parent(s): 64bbed5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ if os.environ.get("SETUP_DONE") != "1":
9
  import spaces
10
 
11
 
12
- #os.environ["JAX_PLATFORMS"] = "cpu"
13
  import gradio as gr
14
  import logging
15
  import sys
@@ -244,7 +244,7 @@ def extract_video_frames(video_path: str):
244
  log.info(f"clip_chunk: {clip_chunk.shape}, sync_chunk: {sync_chunk.shape}")
245
  return clip_chunk, sync_chunk, duration_sec
246
 
247
- @spaces.GPU
248
  def extract_features_cpu(clip_chunk, sync_chunk, caption):
249
  model = _MODELS["feature_extractor"]
250
 
 
9
  import spaces
10
 
11
 
12
+ os.environ["JAX_PLATFORMS"] = "cpu"
13
  import gradio as gr
14
  import logging
15
  import sys
 
244
  log.info(f"clip_chunk: {clip_chunk.shape}, sync_chunk: {sync_chunk.shape}")
245
  return clip_chunk, sync_chunk, duration_sec
246
 
247
+
248
  def extract_features_cpu(clip_chunk, sync_chunk, caption):
249
  model = _MODELS["feature_extractor"]
250