Spaces:
Sleeping
Sleeping
Commit
·
7e7870c
1
Parent(s):
2b228e6
load model checkpoint
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ import matplotlib.ticker as ticker
|
|
| 14 |
from huggingface_hub import HfApi
|
| 15 |
import gradio as gr
|
| 16 |
|
|
|
|
| 17 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 18 |
|
| 19 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 14 |
from huggingface_hub import HfApi
|
| 15 |
import gradio as gr
|
| 16 |
|
| 17 |
+
|
| 18 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 19 |
|
| 20 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|