Spaces:
Sleeping
Sleeping
Update difpoint/src/models/predictor.py
Browse files
difpoint/src/models/predictor.py
CHANGED
|
@@ -10,8 +10,6 @@ from torch.cuda import nvtx
|
|
| 10 |
from collections import OrderedDict
|
| 11 |
import platform
|
| 12 |
|
| 13 |
-
import spaces
|
| 14 |
-
|
| 15 |
try:
|
| 16 |
import tensorrt as trt
|
| 17 |
import ctypes
|
|
@@ -40,7 +38,6 @@ class TensorRTPredictor:
|
|
| 40 |
"""
|
| 41 |
Implements inference for the EfficientDet TensorRT engine.
|
| 42 |
"""
|
| 43 |
-
@spaces.GPU
|
| 44 |
def __init__(self, **kwargs):
|
| 45 |
"""
|
| 46 |
:param engine_path: The path to the serialized engine to load from disk.
|
|
|
|
| 10 |
from collections import OrderedDict
|
| 11 |
import platform
|
| 12 |
|
|
|
|
|
|
|
| 13 |
try:
|
| 14 |
import tensorrt as trt
|
| 15 |
import ctypes
|
|
|
|
| 38 |
"""
|
| 39 |
Implements inference for the EfficientDet TensorRT engine.
|
| 40 |
"""
|
|
|
|
| 41 |
def __init__(self, **kwargs):
|
| 42 |
"""
|
| 43 |
:param engine_path: The path to the serialized engine to load from disk.
|