Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ if os.environ.get("SETUP_DONE") != "1":
|
|
| 9 |
import spaces
|
| 10 |
|
| 11 |
|
| 12 |
-
|
| 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 |
-
|
| 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 |
|