Spaces:
Sleeping
Sleeping
fix types
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
|
|
28 |
@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
29 |
def process_image(
|
30 |
image_input, text_input
|
31 |
-
) -> List[Image]:
|
32 |
if not image_input:
|
33 |
gr.Info("Please upload an image.")
|
34 |
return []
|
|
|
28 |
@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
29 |
def process_image(
|
30 |
image_input, text_input
|
31 |
+
) -> List[Image.Image]:
|
32 |
if not image_input:
|
33 |
gr.Info("Please upload an image.")
|
34 |
return []
|