HoiAlice
commited on
Commit
·
648e439
1
Parent(s):
2afe35a
mem
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ from typing import List, Dict
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 7 |
|
| 8 |
@st.cache_resource
|
| 9 |
def load_model():
|
|
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 7 |
+
# как же я это все ненавижу
|
| 8 |
|
| 9 |
@st.cache_resource
|
| 10 |
def load_model():
|