Spaces:
Build error
Build error
Ubuntu
commited on
Commit
·
c4fa864
1
Parent(s):
0f0d8fb
- .ipynb_checkpoints/app-checkpoint.py +2 -0
- app.py +2 -0
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -7,6 +7,8 @@ import numpy as np
|
|
| 7 |
from torchvision.datasets import ImageFolder
|
| 8 |
from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
|
| 9 |
from pathlib import Path
|
|
|
|
|
|
|
| 10 |
|
| 11 |
def video_identity(video,user_name,class_name,trainortest,ready):
|
| 12 |
if ready=='yes':
|
|
|
|
| 7 |
from torchvision.datasets import ImageFolder
|
| 8 |
from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
|
| 9 |
from pathlib import Path
|
| 10 |
+
import pytorch_lightning as pl
|
| 11 |
+
|
| 12 |
|
| 13 |
def video_identity(video,user_name,class_name,trainortest,ready):
|
| 14 |
if ready=='yes':
|
app.py
CHANGED
|
@@ -7,6 +7,8 @@ import numpy as np
|
|
| 7 |
from torchvision.datasets import ImageFolder
|
| 8 |
from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
|
| 9 |
from pathlib import Path
|
|
|
|
|
|
|
| 10 |
|
| 11 |
def video_identity(video,user_name,class_name,trainortest,ready):
|
| 12 |
if ready=='yes':
|
|
|
|
| 7 |
from torchvision.datasets import ImageFolder
|
| 8 |
from transformers import ViTFeatureExtractor, ViTForImageClassification, AutoFeatureExtractor, ViTMSNForImageClassification
|
| 9 |
from pathlib import Path
|
| 10 |
+
import pytorch_lightning as pl
|
| 11 |
+
|
| 12 |
|
| 13 |
def video_identity(video,user_name,class_name,trainortest,ready):
|
| 14 |
if ready=='yes':
|