Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,8 +79,11 @@ torch.cuda.empty_cache()
|
|
| 79 |
#nvidia-smi
|
| 80 |
|
| 81 |
#pip install wandb
|
|
|
|
| 82 |
import wandb
|
| 83 |
-
|
|
|
|
|
|
|
| 84 |
from transformers import MT5ForConditionalGeneration, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
| 85 |
import torch
|
| 86 |
|
|
|
|
| 79 |
#nvidia-smi
|
| 80 |
|
| 81 |
#pip install wandb
|
| 82 |
+
import os
|
| 83 |
import wandb
|
| 84 |
+
print(os.getenv('API_KEY'))
|
| 85 |
+
|
| 86 |
+
wandb.login(key="API_KEY")
|
| 87 |
from transformers import MT5ForConditionalGeneration, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
| 88 |
import torch
|
| 89 |
|