a

.ipynb_checkpoints/app-checkpoint.py

@@ -9,6 +9,7 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFe
 from pathlib import Path
 import pytorch_lightning as pl
 from torch.utils.data import DataLoader
+from torchmetrics import Accuracy
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':

app.py

@@ -9,6 +9,7 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFe
 from pathlib import Path
 import pytorch_lightning as pl
 from torch.utils.data import DataLoader
+from torchmetrics import Accuracy
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':