PDG commited on
Commit
8241067
·
1 Parent(s): 574428c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def classifyCar(im):
76
  #with torch.no_grad():
77
  # scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
78
  #{LABELS[i]: float(scores[i]) for i in range(n_labels)}
79
- return Image.fromarray(np.uint8(out.get_image())).convert('RGB'), "test"
 
80
 
81
  #examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
82
 
 
76
  #with torch.no_grad():
77
  # scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
78
  #{LABELS[i]: float(scores[i]) for i in range(n_labels)}
79
+ #Image.fromarray(np.uint8(out.get_image())).convert('RGB')
80
+ return im, "test"
81
 
82
  #examples = [[example_img.jpg], [example_img2.jpg]] # must be uploaded in repo
83