kiranbandi commited on
Commit
d44e86a
·
verified ·
1 Parent(s): e35d20f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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]