PDG commited on
Commit
316533c
·
1 Parent(s): e668ea9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ def classifyCar(im):
93
  # scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
94
  #{LABELS[i]: float(scores[i]) for i in range(n_labels)}
95
  #Image.fromarray(np.uint8(out.get_image())).convert('RGB')
96
- return label, im2
97
 
98
  #examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
99
 
 
93
  # scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
94
  #{LABELS[i]: float(scores[i]) for i in range(n_labels)}
95
  #Image.fromarray(np.uint8(out.get_image())).convert('RGB')
96
+ return im2, label
97
 
98
  #examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
99