karths commited on
Commit
58df6b7
·
verified ·
1 Parent(s): d964496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -12,6 +12,9 @@ from threading import Thread
12
  from collections.abc import Iterator
13
  import csv
14
 
 
 
 
15
  # Increase CSV field size limit
16
  csv.field_size_limit(1000000)
17
 
@@ -26,9 +29,7 @@ torch.manual_seed(seed)
26
  if torch.cuda.is_available():
27
  torch.cuda.manual_seed_all(seed)
28
 
29
- # Login to Hugging Face
30
- token = os.getenv("hf_token")
31
- login(token)
32
 
33
  model_paths = [
34
  'karths/binary_classification_train_port',
 
12
  from collections.abc import Iterator
13
  import csv
14
 
15
+ # Login to Hugging Face
16
+ token = os.getenv("hf_token")
17
+ login(token=token)
18
  # Increase CSV field size limit
19
  csv.field_size_limit(1000000)
20
 
 
29
  if torch.cuda.is_available():
30
  torch.cuda.manual_seed_all(seed)
31
 
32
+
 
 
33
 
34
  model_paths = [
35
  'karths/binary_classification_train_port',