Update app.py

app.py

@@ -19,12 +19,18 @@ pipe = pipeline('object-detection', model=model, feature_extractor=extractor)
 
 img_url = st.text_input('Image URL', 'https://images.unsplash.com/photo-1556911220-bff31c812dba?ixlib=rb-4.0.3&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=2468&q=80')
 
+st.caption('Downloading Image...')
+
 img_data = requests.get(img_url).content
 with open('detect.jpg', 'wb') as handler:
     handler.write(img_data)
 
+st.caption('Running Detection...')
+
 output = pipe(img_url)
 
+st.caption('Adding Predictions to Image...')
+
 fpath = "Poppins-SemiBold.ttf"
 prop = font_manager.FontProperties(fname=fpath)
 
@@ -55,4 +61,6 @@ image = Image.open('detect-bbox.jpg')
 
 st.image(image, caption='DETR Image')
 
-plt.show()
+plt.show()
+
+st.caption('Done!')