Bijan k
commited on
Commit
•
83c18d3
1
Parent(s):
4820778
testing
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ model.load_state_dict(
|
|
| 32 |
|
| 33 |
def preprocessImg(img):
|
| 34 |
transform = transforms.Compose([
|
| 35 |
-
|
| 36 |
transforms.Resize((256,256)),
|
| 37 |
transforms.ToTensor(),
|
| 38 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
|
|
|
| 32 |
|
| 33 |
def preprocessImg(img):
|
| 34 |
transform = transforms.Compose([
|
| 35 |
+
# transforms.Grayscale(),
|
| 36 |
transforms.Resize((256,256)),
|
| 37 |
transforms.ToTensor(),
|
| 38 |
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|