Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import os
 from matplotlib import rcParams, font_manager
 import streamlit as st
+import urllib.request
 
 extractor = AutoFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
 
@@ -19,10 +20,12 @@ img_url = st.text_input('Image URL', 'https://images.unsplash.com/photo-15569112
 
 output = pipe(img_url)
 
+urllib.request.urlretrieve(img_url, "detect.jpg")
+
 fpath = os.path.join(r"Poppins-SemiBold.ttf")
 prop = font_manager.FontProperties(fname=fpath)
 
-img = Image.open('kitchen.jpg')
+img = Image.open('detect.jpg')
 plt.figure(dpi=2400)
 
 # Create figure and axes
@@ -43,9 +46,9 @@ for prediction in output:
 
 plt.axis('off')
 
-plt.savefig('kitchen-bbox.jpg', dpi=1200, bbox_inches='tight')
+plt.savefig('detect-bbox.jpg', dpi=1200, bbox_inches='tight')
 
-image = Image.open('kitchen-bbox.jpg')
+image = Image.open('detect-bbox.jpg')
 
 st.image(image, caption='DETR Image')