nickmuchi commited on
Commit
ade62d9
1 Parent(s): ec26874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,8 +68,8 @@ def detect_objects(model_name,url_input,image_input,threshold):
68
  if url_input and validators.url(url_input):
69
  image = Image.open(requests.get(url_input, stream=True).raw)
70
 
71
- elif image_upload:
72
- image = image_upload
73
 
74
  #Make prediction
75
  processed_outputs = make_prediction(image, feature_extractor, model)
 
68
  if url_input and validators.url(url_input):
69
  image = Image.open(requests.get(url_input, stream=True).raw)
70
 
71
+ elif image_input:
72
+ image = image_input
73
 
74
  #Make prediction
75
  processed_outputs = make_prediction(image, feature_extractor, model)