Spaces:
Build error
Build error
arxiv id list support
Browse files- app.py +0 -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 |
-
|
91 |
-
|
92 |
-
|
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"]
|