Spaces:
Runtime error
Runtime error
Commit
•
773b816
1
Parent(s):
e81673f
check torch version
Browse files
app.py
CHANGED
|
@@ -6,6 +6,8 @@ import pandas as pd
|
|
| 6 |
|
| 7 |
SEED = 42
|
| 8 |
|
|
|
|
|
|
|
| 9 |
@st.cache_resource
|
| 10 |
def init_count_model():
|
| 11 |
return torch.load("count_probs.pt")
|
|
|
|
| 6 |
|
| 7 |
SEED = 42
|
| 8 |
|
| 9 |
+
print(f"torch version -----> {torch.__version__}")
|
| 10 |
+
|
| 11 |
@st.cache_resource
|
| 12 |
def init_count_model():
|
| 13 |
return torch.load("count_probs.pt")
|