SamDaLamb commited on
Commit
b74b614
·
verified ·
1 Parent(s): 9aa606e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -37,7 +37,7 @@ model.eval()
37
  # transforms.ToTensor()
38
  # ])
39
  transform = transforms.Compose([ # Ensure input is a PIL image
40
- transforms.Resize((640, 640)),
41
  transforms.ToTensor()
42
  ])
43
  # transform = transforms.Compose([
@@ -85,6 +85,7 @@ def detect_objects_in_image(image):
85
  print(Image.fromarray(np.array(image)))
86
  print(np.array(image))
87
  print(type(image))
 
88
  if len(pred) == 0:
89
  return Image.fromarray(np.array(image)) # Return only image and None for graph
90
  print("passed6")
 
37
  # transforms.ToTensor()
38
  # ])
39
  transform = transforms.Compose([ # Ensure input is a PIL image
40
+ transforms.Resize((1000, 1000)),
41
  transforms.ToTensor()
42
  ])
43
  # transform = transforms.Compose([
 
85
  print(Image.fromarray(np.array(image)))
86
  print(np.array(image))
87
  print(type(image))
88
+ print(len(pred))
89
  if len(pred) == 0:
90
  return Image.fromarray(np.array(image)) # Return only image and None for graph
91
  print("passed6")