ayaanzaveri commited on
Commit
44aa716
1 Parent(s): 912906e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -7,6 +7,7 @@ import os
7
  from matplotlib import rcParams, font_manager
8
  import streamlit as st
9
  import urllib.request
 
10
 
11
  extractor = AutoFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
12
 
@@ -18,11 +19,13 @@ pipe = pipeline('object-detection', model=model, feature_extractor=extractor)
18
 
19
  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')
20
 
21
- output = pipe(img_url)
 
 
22
 
23
- urllib.request.urlretrieve(img_url, "detect.jpg")
24
 
25
- fpath = os.path.join(r"Poppins-SemiBold.ttf")
26
  prop = font_manager.FontProperties(fname=fpath)
27
 
28
  img = Image.open('detect.jpg')
 
7
  from matplotlib import rcParams, font_manager
8
  import streamlit as st
9
  import urllib.request
10
+ import requests
11
 
12
  extractor = AutoFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
13
 
 
19
 
20
  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')
21
 
22
+ img_data = requests.get(image_url).content
23
+ with open('detect.jpg', 'wb') as handler:
24
+ handler.write(img_data)
25
 
26
+ output = pipe(img_url)
27
 
28
+ fpath = "Poppins-SemiBold.ttf"
29
  prop = font_manager.FontProperties(fname=fpath)
30
 
31
  img = Image.open('detect.jpg')