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