Spaces:
Build error
Build error
Update app.py
Browse files
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'))
|