sidphbot commited on
Commit
48cd509
1 Parent(s): f310b8b

arxiv id list support

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  2. src/Surveyor.py +3 -3
app.py CHANGED
@@ -1,5 +1,3 @@
1
-
2
- from turtle import down
3
  import streamlit as st
4
  import pandas as pd
5
  import numpy as np
 
 
 
1
  import streamlit as st
2
  import pandas as pd
3
  import numpy as np
src/Surveyor.py CHANGED
@@ -87,9 +87,9 @@ class Surveyor:
87
  spacy.require_gpu()
88
 
89
  self.high_gpu = high_gpu
90
- #self.DEFAULTS = DEFAULTS_CPU_COMPAT
91
- #if self.high_gpu:
92
- self.DEFAULTS = DEFAULTS_HIGH_GPU
93
 
94
  if not kw_model_name:
95
  kw_model_name = self.DEFAULTS["kw_model_name"]
 
87
  spacy.require_gpu()
88
 
89
  self.high_gpu = high_gpu
90
+ self.DEFAULTS = DEFAULTS_CPU_COMPAT
91
+ if self.high_gpu:
92
+ self.DEFAULTS = DEFAULTS_HIGH_GPU
93
 
94
  if not kw_model_name:
95
  kw_model_name = self.DEFAULTS["kw_model_name"]