Hugging Face's logo

Spaces:
AlicanA
/
Custom-Object-Detection
Build error

App Files Community
Ubuntu commited on
Commit
0f0d8fb
·
1 Parent(s): 55d1370

a

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -21,7 +21,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
21
  label2id = {}
22
  id2label = {}
23
 
24
- for i, class_name in enumerate(ds.classes):
25
  label2id[class_name] = str(i)
26
  id2label[str(i)] = class_name
27
 
 
21
  label2id = {}
22
  id2label = {}
23
 
24
+ for i, class_name in enumerate(train_ds.classes):
25
  label2id[class_name] = str(i)
26
  id2label[str(i)] = class_name
27
 
app.py CHANGED
@@ -21,7 +21,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
21
  label2id = {}
22
  id2label = {}
23
 
24
- for i, class_name in enumerate(ds.classes):
25
  label2id[class_name] = str(i)
26
  id2label[str(i)] = class_name
27
 
 
21
  label2id = {}
22
  id2label = {}
23
 
24
+ for i, class_name in enumerate(train_ds.classes):
25
  label2id[class_name] = str(i)
26
  id2label[str(i)] = class_name
27