SamDaLamb commited on
Commit
86c7aca
·
verified ·
1 Parent(s): d2545e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ OBJECT_NAMES = ['enemies']
51
  def detect_objects_in_image(image):
52
  # Ensure image is a PIL Image
53
  image = Image.fromarray(image)
54
- data = Image.fromarray(image)
55
  if isinstance(image, torch.Tensor):
56
  image = transforms.ToPILImage()(image) # Convert tensor to PIL image
57
 
 
51
  def detect_objects_in_image(image):
52
  # Ensure image is a PIL Image
53
  image = Image.fromarray(image)
54
+
55
  if isinstance(image, torch.Tensor):
56
  image = transforms.ToPILImage()(image) # Convert tensor to PIL image
57