Update app.py
Browse files
app.py
CHANGED
|
@@ -319,7 +319,7 @@ def clear_ref_clicks(ref_state):
|
|
| 319 |
return Image.fromarray(ref_state["origin_image"])
|
| 320 |
|
| 321 |
|
| 322 |
-
@spaces.GPU(
|
| 323 |
def track_video(n_frames, video_state):
|
| 324 |
input_points = video_state["input_points"]
|
| 325 |
input_labels = video_state["input_labels"]
|
|
@@ -398,7 +398,7 @@ def track_video(n_frames, video_state):
|
|
| 398 |
return video_file, video_state
|
| 399 |
|
| 400 |
|
| 401 |
-
@spaces.GPU(
|
| 402 |
def inference_and_return_video(
|
| 403 |
dilate_radius,
|
| 404 |
num_inference_steps,
|
|
|
|
| 319 |
return Image.fromarray(ref_state["origin_image"])
|
| 320 |
|
| 321 |
|
| 322 |
+
@spaces.GPU()
|
| 323 |
def track_video(n_frames, video_state):
|
| 324 |
input_points = video_state["input_points"]
|
| 325 |
input_labels = video_state["input_labels"]
|
|
|
|
| 398 |
return video_file, video_state
|
| 399 |
|
| 400 |
|
| 401 |
+
@spaces.GPU()
|
| 402 |
def inference_and_return_video(
|
| 403 |
dilate_radius,
|
| 404 |
num_inference_steps,
|