SamDaLamb commited on
Commit
b222e8b
·
verified ·
1 Parent(s): 9a57456

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -53,6 +53,9 @@ def detect_objects_in_image(image):
53
  print(np.ndarray.view(image))
54
  img_tensor = transform(image).unsqueeze(0)
55
  print(image.size)
 
 
 
56
  orig_w, orig_h = image.size
57
 
58
 
 
53
  print(np.ndarray.view(image))
54
  img_tensor = transform(image).unsqueeze(0)
55
  print(image.size)
56
+ if isinstance(image, np.ndarray):
57
+ print("Converting NumPy array to PIL Image")
58
+ image = Image.fromarray(image)
59
  orig_w, orig_h = image.size
60
 
61