Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,9 @@ from overview import NQDOverview
|
|
11 |
|
12 |
import torch
|
13 |
|
|
|
|
|
|
|
14 |
cuda_available = torch.cuda.is_available()
|
15 |
print(f"Is CUDA available: {cuda_available}")
|
16 |
if cuda_available:
|
@@ -35,7 +38,6 @@ def get_results(model, c):
|
|
35 |
return {'label': label, 'scores': scores}
|
36 |
|
37 |
def run_models(model_names, models, c):
|
38 |
-
print(c)
|
39 |
results = {}
|
40 |
for mn in model_names:
|
41 |
results[mn] = get_results(models[mn], c)
|
@@ -73,7 +75,6 @@ lc_placeholder.empty()
|
|
73 |
|
74 |
### Load example data
|
75 |
examples = load_pickle('test.pkl')
|
76 |
-
print(examples)
|
77 |
|
78 |
### Process input
|
79 |
ex = examples['comment'].sample(1, random_state=int(datetime.now().timestamp())).tolist()[0]
|
|
|
11 |
|
12 |
import torch
|
13 |
|
14 |
+
import pkg_resources
|
15 |
+
print([p.project_name for p in pkg_resources.working_set])
|
16 |
+
|
17 |
cuda_available = torch.cuda.is_available()
|
18 |
print(f"Is CUDA available: {cuda_available}")
|
19 |
if cuda_available:
|
|
|
38 |
return {'label': label, 'scores': scores}
|
39 |
|
40 |
def run_models(model_names, models, c):
|
|
|
41 |
results = {}
|
42 |
for mn in model_names:
|
43 |
results[mn] = get_results(models[mn], c)
|
|
|
75 |
|
76 |
### Load example data
|
77 |
examples = load_pickle('test.pkl')
|
|
|
78 |
|
79 |
### Process input
|
80 |
ex = examples['comment'].sample(1, random_state=int(datetime.now().timestamp())).tolist()[0]
|