Spaces:
Sleeping
Sleeping
Commit
·
deb5853
1
Parent(s):
1468b43
Minor bug fixes
Browse files- app.py +3 -0
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,11 +1,14 @@
|
|
| 1 |
import wandb
|
| 2 |
import torch
|
| 3 |
import re
|
|
|
|
| 4 |
|
| 5 |
import gradio
|
| 6 |
|
| 7 |
from transformers import GPT2Tokenizer,GPT2LMHeadModel
|
| 8 |
|
|
|
|
|
|
|
| 9 |
wandb.login()
|
| 10 |
|
| 11 |
run = wandb.init(project="Question_Answer", job_type="model_loading", id='xeew4vz7', resume="must")
|
|
|
|
| 1 |
import wandb
|
| 2 |
import torch
|
| 3 |
import re
|
| 4 |
+
import os
|
| 5 |
|
| 6 |
import gradio
|
| 7 |
|
| 8 |
from transformers import GPT2Tokenizer,GPT2LMHeadModel
|
| 9 |
|
| 10 |
+
os.environ["WANDB_API_KEY"] = "d2ad0a7285379c0808ca816971d965fc242d0b5e"
|
| 11 |
+
|
| 12 |
wandb.login()
|
| 13 |
|
| 14 |
run = wandb.init(project="Question_Answer", job_type="model_loading", id='xeew4vz7', resume="must")
|
requirements.txt
CHANGED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
wandb
|