SamDaLamb commited on
Commit
5ff0a7b
·
verified ·
1 Parent(s): f6679f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -50,9 +50,7 @@ transform = transforms.Compose([
50
  OBJECT_NAMES = ['enemies']
51
 
52
  def detect_objects_in_image(image):
53
- """
54
- Detect objects in the given image.
55
- """
56
  # Ensure image is a PIL Image
57
  if isinstance(image, torch.Tensor):
58
  image = transforms.ToPILImage()(image) # Convert tensor to PIL image
 
50
  OBJECT_NAMES = ['enemies']
51
 
52
  def detect_objects_in_image(image):
53
+
 
 
54
  # Ensure image is a PIL Image
55
  if isinstance(image, torch.Tensor):
56
  image = transforms.ToPILImage()(image) # Convert tensor to PIL image