Spaces:
Sleeping
Sleeping
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"))
|