Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,6 @@ import gradio as gr
|
|
| 3 |
import os
|
| 4 |
os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "1"
|
| 5 |
import torch
|
| 6 |
-
import spaces
|
| 7 |
import numpy as np
|
| 8 |
import cv2
|
| 9 |
import matplotlib.pyplot as plt
|
|
@@ -130,7 +129,7 @@ def show_masks(image, masks, scores, point_coords=None, box_coords=None, input_l
|
|
| 130 |
|
| 131 |
return combined_images, mask_images
|
| 132 |
|
| 133 |
-
|
| 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"))
|
|
|
|
| 3 |
import os
|
| 4 |
os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "1"
|
| 5 |
import torch
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
import cv2
|
| 8 |
import matplotlib.pyplot as plt
|
|
|
|
| 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"))
|