SkalskiP commited on
Commit
2c71d17
·
1 Parent(s): b32b0a3
Files changed (1) hide show
  1. app.py +1 -1
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 []