AlicanA commited on
Commit
52c695a
·
1 Parent(s): 949b93c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
74
 
75
 
76
  train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=8, shuffle=True)
77
- test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=2)
78
 
79
 
80
  for name, param in model.named_parameters():
@@ -131,4 +131,4 @@ demo = gr.Interface(video_identity,
131
  gr.Text(),
132
  gr.Text()],
133
  cache_examples=True)
134
- demo.launch(debug=True)
 
74
 
75
 
76
  train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=8, shuffle=True)
77
+ test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=8)
78
 
79
 
80
  for name, param in model.named_parameters():
 
131
  gr.Text(),
132
  gr.Text()],
133
  cache_examples=True)
134
+ demo.launch(debug=True)