HugMoE commited on
Commit
692fd2b
1 Parent(s): 60c7ee2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -33,6 +33,8 @@ def predict(image, threshold=0.25, model_id=None):
33
  results = model(image, size=input_size)
34
  numpy_image = results.render()[0]
35
  output_image = Image.fromarray(numpy_image)
 
 
36
  return output_image
37
 
38
 
 
33
  results = model(image, size=input_size)
34
  numpy_image = results.render()[0]
35
  output_image = Image.fromarray(numpy_image)
36
+ predictions = results.pred[0]
37
+ print(predictions[:, :4])
38
  return output_image
39
 
40