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"]
|