Spaces:

spdraptor
/

Photo_edit

Paused

spdraptor commited on Aug 30, 2024

Commit

8b87815

1 Parent(s): eebe067

fast api cod commit

Files changed (2) hide show

modules/__pycache__/masking_module.cpython-310.pyc CHANGED Viewed

Binary files a/modules/__pycache__/masking_module.cpython-310.pyc and b/modules/__pycache__/masking_module.cpython-310.pyc differ

modules/masking_module.py CHANGED Viewed

@@ -165,10 +165,11 @@ def masking_process(image,obj):
     response = run_sam_inference(SAM_IMAGE_MODEL, image, detections)
     print('5')
     if response['code'] == 400:
-      return response
     else:
-      detections2=response['data']
-    mask = Image.fromarray(detections2.mask[0])
-    response['data']=mask
-    torch.cuda.empty_cache()
-    return response

     response = run_sam_inference(SAM_IMAGE_MODEL, image, detections)
     print('5')
     if response['code'] == 400:
+        print("no object found")
+        return "no object found"
     else:
+        detections2=response['data']
+        mask = Image.fromarray(detections2.mask[0])
+        # response['data']=mask
+        torch.cuda.empty_cache()
+        return mask