Spaces:
Running
on
Zero
Running
on
Zero
ZeroGPU ready
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
import os
|
|
@@ -129,7 +130,7 @@ def show_masks(image, masks, scores, point_coords=None, box_coords=None, input_l
|
|
| 129 |
|
| 130 |
return combined_images, mask_images
|
| 131 |
|
| 132 |
-
|
| 133 |
def sam_process(input_image, checkpoint, tracking_points, trackings_input_label):
|
| 134 |
image = Image.open(input_image)
|
| 135 |
image = np.array(image.convert("RGB"))
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
import os
|
|
|
|
| 130 |
|
| 131 |
return combined_images, mask_images
|
| 132 |
|
| 133 |
+
@spaces.GPU
|
| 134 |
def sam_process(input_image, checkpoint, tracking_points, trackings_input_label):
|
| 135 |
image = Image.open(input_image)
|
| 136 |
image = np.array(image.convert("RGB"))
|