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 |
|