Tao Wu commited on
Commit
a70f9b1
1 Parent(s): 5f285e1
Files changed (1) hide show
  1. app/embedding_setup.py +1 -1
app/embedding_setup.py CHANGED
@@ -9,7 +9,7 @@ import os
9
  import torch
10
  from predibase import Predibase, FinetuningConfig, DeploymentConfig
11
  pb_auth = os.environ.get("pb_token")
12
- pb = Predibase(api_token=pb_auth)
13
 
14
  if torch.cuda.is_available():
15
  device = "cuda"
 
9
  import torch
10
  from predibase import Predibase, FinetuningConfig, DeploymentConfig
11
  pb_auth = os.environ.get("pb_token")
12
+ pb = Predibase(api_token='pb_8PhUywZ5VQCQFmGl8ZgfiQ')
13
 
14
  if torch.cuda.is_available():
15
  device = "cuda"