Update app.py

app.py

@@ -17,21 +17,19 @@ from transformers import pipeline
 
 pipe = pipeline('object-detection', model=model, feature_extractor=extractor)
 
-my_bar = st.progress(0)
-
 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')
 
-my_bar.progress(20)
+st.caption('Downloading Image...')
 
 img_data = requests.get(img_url).content
 with open('detect.jpg', 'wb') as handler:
     handler.write(img_data)
 
-my_bar.progress(60)
+st.caption('Running Detection...')
 
 output = pipe(img_url)
 
-my_bar.progress(80)
+st.caption('Adding Predictions to Image...')
 
 fpath = "Poppins-SemiBold.ttf"
 prop = font_manager.FontProperties(fname=fpath)
@@ -65,10 +63,4 @@ st.image(image, caption='DETR Image')
 
 plt.show()
 
-my_bar.progress(100)
-
-st.download_button(
-    label="Download Image",
-    data="detect-bbox.jpg",
-    file_name='detect-bbox.jpg',
-)
+st.caption('Done!')