Mayanand commited on
Commit
de12553
·
1 Parent(s): ee35021

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def predict_fn(image, link):
40
  error_text = 'Image from given link could not be downloaded, please try again with valid link'
41
  return error_image, error_text
42
 
43
- display_image = image
44
  image = val_tfms(image).unsqueeze(0)
45
  model.eval()
46
  out = model.predict(image, torch.device('cpu'))
 
40
  error_text = 'Image from given link could not be downloaded, please try again with valid link'
41
  return error_image, error_text
42
 
43
+ display_image = transforms.Resize(100)(image)
44
  image = val_tfms(image).unsqueeze(0)
45
  model.eval()
46
  out = model.predict(image, torch.device('cpu'))