Hugging Face's logo

Spaces:
envision-ai
/
Custom-Object-Detection
Runtime error

App Files Community
Ubuntu commited on
Commit
dbcdf98
•
1 Parent(s): 9aafa24

a

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +2 -2
  2. app.py +2 -2
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -56,11 +56,11 @@ def video_identity(video,user_name,class_name,trainortest,ready):
56
 
57
  data_dir = Path(str(user_name)+'/train')
58
 
59
- train_ds = ImageFolder(data_dir, transform=transform)
60
 
61
 
62
  test_dir = Path(str(user_name)+'/test')
63
- test_ds = ImageFolder(test_dir, transform=transform)
64
 
65
  label2id = {}
66
  id2label = {}
 
56
 
57
  data_dir = Path(str(user_name)+'/train')
58
 
59
+ train_ds = ImageFolder(data_dir)
60
 
61
 
62
  test_dir = Path(str(user_name)+'/test')
63
+ test_ds = ImageFolder(test_dir)
64
 
65
  label2id = {}
66
  id2label = {}
app.py CHANGED
@@ -56,11 +56,11 @@ def video_identity(video,user_name,class_name,trainortest,ready):
56
 
57
  data_dir = Path(str(user_name)+'/train')
58
 
59
- train_ds = ImageFolder(data_dir, transform=transform)
60
 
61
 
62
  test_dir = Path(str(user_name)+'/test')
63
- test_ds = ImageFolder(test_dir, transform=transform)
64
 
65
  label2id = {}
66
  id2label = {}
 
56
 
57
  data_dir = Path(str(user_name)+'/train')
58
 
59
+ train_ds = ImageFolder(data_dir)
60
 
61
 
62
  test_dir = Path(str(user_name)+'/test')
63
+ test_ds = ImageFolder(test_dir)
64
 
65
  label2id = {}
66
  id2label = {}