Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -240,7 +240,7 @@ def preprocess_for_removal(images, masks):
|
|
| 240 |
MAX_PROC_SIDE = None # None = native resolution; set e.g. 1280 if you hit OOM
|
| 241 |
|
| 242 |
@spaces.GPU(duration=300) # 1080p is heavy — give it time
|
| 243 |
-
def inference_and_return_video(dilation_iterations, num_inference_steps, video_state):
|
| 244 |
if video_state.get("video_path") is None:
|
| 245 |
video_state["video_path"] = video_path
|
| 246 |
if video_state["origin_images"] is None or video_state["masks"] is None:
|
|
|
|
| 240 |
MAX_PROC_SIDE = None # None = native resolution; set e.g. 1280 if you hit OOM
|
| 241 |
|
| 242 |
@spaces.GPU(duration=300) # 1080p is heavy — give it time
|
| 243 |
+
def inference_and_return_video(dilation_iterations, num_inference_steps, video_path, video_state):
|
| 244 |
if video_state.get("video_path") is None:
|
| 245 |
video_state["video_path"] = video_path
|
| 246 |
if video_state["origin_images"] is None or video_state["masks"] is None:
|