drsaikirant88 commited on
Commit
1035ec8
1 Parent(s): 74808e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -137,7 +137,7 @@ def detector(image):
137
  try:
138
  output
139
  except NameError:
140
- return None
141
 
142
  im_dim_list = torch.index_select(im_dim_list, 0, output[:,0].long())
143
 
 
137
  try:
138
  output
139
  except NameError:
140
+ return loaded_ims[0], []
141
 
142
  im_dim_list = torch.index_select(im_dim_list, 0, output[:,0].long())
143