SWHL commited on
Commit
78410a0
1 Parent(s): 9b5cb05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -80,7 +80,8 @@ img_file_buffer = st.file_uploader(
80
  if img_file_buffer:
81
  image = Image.open(img_file_buffer)
82
  img = np.array(image)
83
- boxes, scores, class_names, *elapse = layout_engine(img)
84
- ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
 
85
 
86
  st.image(ploted_img, use_column_width=True)
 
80
  if img_file_buffer:
81
  image = Image.open(img_file_buffer)
82
  img = np.array(image)
83
+ with st.spinner("推理中...."):
84
+ boxes, scores, class_names, *elapse = layout_engine(img)
85
+ ploted_img = VisLayout.draw_detections(img, boxes, scores, class_names)
86
 
87
  st.image(ploted_img, use_column_width=True)