Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,7 @@ def preprocess_image(image):
|
|
| 65 |
img = img.unsqueeze(0)
|
| 66 |
print("in preprocess-2 img.shape:",img.shape)
|
| 67 |
|
| 68 |
-
return img,
|
| 69 |
|
| 70 |
def infer(model, img):
|
| 71 |
with torch.no_grad():
|
|
|
|
| 65 |
img = img.unsqueeze(0)
|
| 66 |
print("in preprocess-2 img.shape:",img.shape)
|
| 67 |
|
| 68 |
+
return img, image
|
| 69 |
|
| 70 |
def infer(model, img):
|
| 71 |
with torch.no_grad():
|