Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
·
e8e2886
1
Parent(s):
2911751
init
Browse files- .ipynb_checkpoints/app-checkpoint.py +1 -1
- app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -19,7 +19,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 19 |
|
| 20 |
data_dir = Path(str(user_name)+'/train')
|
| 21 |
transform = transforms.Compose([
|
| 22 |
-
transforms.
|
| 23 |
])
|
| 24 |
train_ds = ImageFolder(data_dir, transform=transform)
|
| 25 |
|
|
|
|
| 19 |
|
| 20 |
data_dir = Path(str(user_name)+'/train')
|
| 21 |
transform = transforms.Compose([
|
| 22 |
+
transforms.ConvertImageDtype(torch.float32),
|
| 23 |
])
|
| 24 |
train_ds = ImageFolder(data_dir, transform=transform)
|
| 25 |
|
app.py
CHANGED
|
@@ -19,7 +19,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 19 |
|
| 20 |
data_dir = Path(str(user_name)+'/train')
|
| 21 |
transform = transforms.Compose([
|
| 22 |
-
transforms.
|
| 23 |
])
|
| 24 |
train_ds = ImageFolder(data_dir, transform=transform)
|
| 25 |
|
|
|
|
| 19 |
|
| 20 |
data_dir = Path(str(user_name)+'/train')
|
| 21 |
transform = transforms.Compose([
|
| 22 |
+
transforms.ConvertImageDtype(torch.float32),
|
| 23 |
])
|
| 24 |
train_ds = ImageFolder(data_dir, transform=transform)
|
| 25 |
|