Abdulmohsena commited on
Commit
27d22c2
·
verified ·
1 Parent(s): d7a2bf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -4,7 +4,10 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
  import torch
5
  import warnings
6
  import weave
 
 
7
 
 
8
  weave.init("Faseeh")
9
 
10
  warnings.filterwarnings("ignore")
 
4
  import torch
5
  import warnings
6
  import weave
7
+ import os
8
+ import wandb
9
 
10
+ wandb.login(key=os.getenv('WANDB_KEY'))
11
  weave.init("Faseeh")
12
 
13
  warnings.filterwarnings("ignore")