Update app.py

app.py

@@ -17,19 +17,21 @@ 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')
 
-st.caption('Downloading Image...')
+my_bar.progress(percent_complete + 20)
 
 img_data = requests.get(img_url).content
 with open('detect.jpg', 'wb') as handler:
     handler.write(img_data)
 
-st.caption('Running Detection...')
+my_bar.progress(percent_complete + 40)
 
 output = pipe(img_url)
 
-st.caption('Adding Predictions to Image...')
+my_bar.progress(percent_complete + 20)
 
 fpath = "Poppins-SemiBold.ttf"
 prop = font_manager.FontProperties(fname=fpath)
@@ -63,4 +65,4 @@ st.image(image, caption='DETR Image')
 
 plt.show()
 
-st.caption('Done!')
+my_bar.progress(percent_complete + 20)