num of workers = 0
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def loading_data(img):
|
|
| 55 |
])
|
| 56 |
# dcreate the dataset
|
| 57 |
test_set = data(img=img, transform=transform)
|
| 58 |
-
test_loader = DataLoader(test_set, batch_size=1, num_workers=
|
| 59 |
|
| 60 |
return test_loader
|
| 61 |
|
|
|
|
| 55 |
])
|
| 56 |
# dcreate the dataset
|
| 57 |
test_set = data(img=img, transform=transform)
|
| 58 |
+
test_loader = DataLoader(test_set, batch_size=1, num_workers=0, shuffle=False, drop_last=False)
|
| 59 |
|
| 60 |
return test_loader
|
| 61 |
|