We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c905222 commit 11ac584Copy full SHA for 11ac584
app_inference_spec.py
@@ -2,8 +2,18 @@
2
from pydantic import BaseModel
3
from typing import List, Tuple
4
import spacy
5
+import logging
6
from models_host.base_inference_spec import BaseInferenceSpec
7
8
+logger = logging.getLogger("competitor-check")
9
+
10
+try:
11
+ spacy.require_gpu()
12
+except Exception as e:
13
+ logger.warning(
14
+ f"Failed to enable GPU ({device}), falling back to CPU: {e}"
15
+ )
16
17
class InferenceData(BaseModel):
18
name: str
19
shape: List[int]
0 commit comments