Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -179,7 +179,6 @@ def preprocess_for_removal(images, masks):
|
|
179 |
arr_masks = np.stack(out_masks)
|
180 |
return torch.from_numpy(arr_images).half().to(device), torch.from_numpy(arr_masks).half().to(device)
|
181 |
|
182 |
-
@spaces.GPU(duration=100)
|
183 |
def inference_and_return_video(dilation_iterations, num_inference_steps, video_state=None):
|
184 |
if video_state["origin_images"] is None or video_state["masks"] is None:
|
185 |
return None
|
@@ -218,7 +217,6 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
|
|
218 |
clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
|
219 |
return video_file
|
220 |
|
221 |
-
@spaces.GPU(duration=100)
|
222 |
def track_video(n_frames, video_state):
|
223 |
|
224 |
input_points = video_state["input_points"]
|
|
|
179 |
arr_masks = np.stack(out_masks)
|
180 |
return torch.from_numpy(arr_images).half().to(device), torch.from_numpy(arr_masks).half().to(device)
|
181 |
|
|
|
182 |
def inference_and_return_video(dilation_iterations, num_inference_steps, video_state=None):
|
183 |
if video_state["origin_images"] is None or video_state["masks"] is None:
|
184 |
return None
|
|
|
217 |
clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
|
218 |
return video_file
|
219 |
|
|
|
220 |
def track_video(n_frames, video_state):
|
221 |
|
222 |
input_points = video_state["input_points"]
|