905saini commited on
Commit
cfad947
·
1 Parent(s): 9c183d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -25,8 +25,9 @@ if file_name is not None:
25
  image = Image.open(file_name)
26
  input_image, copy_img= process_img(image)
27
  col1.image(input_image, use_column_width=True)
28
-
29
- bbox,confidance,classes,nc = model(input_image)
 
30
  bbox,confidance , classes , nc = bbox[0].numpy(),confidance[0].numpy(),classes[0].numpy(),nc[0].numpy()
31
 
32
  for i in range(nc):
 
25
  image = Image.open(file_name)
26
  input_image, copy_img= process_img(image)
27
  col1.image(input_image, use_column_width=True)
28
+ output = model(input_image)
29
+ print(output)
30
+ #bbox,confidance,classes,nc = model(input_image)
31
  bbox,confidance , classes , nc = bbox[0].numpy(),confidance[0].numpy(),classes[0].numpy(),nc[0].numpy()
32
 
33
  for i in range(nc):