Spaces:
Sleeping
Sleeping
Update difpoint/src/models/predictor.py
Browse files
difpoint/src/models/predictor.py
CHANGED
@@ -265,7 +265,6 @@ class OnnxRuntimePredictorSingleton(OnnxRuntimePredictor):
|
|
265 |
#return OnnxRuntimePredictorSingleton._instance[model_path]
|
266 |
return OnnxRuntimePredictor(**kwargs)
|
267 |
|
268 |
-
|
269 |
def get_predictor(**kwargs):
|
270 |
predict_type = kwargs.get("predict_type", "trt")
|
271 |
if predict_type == "ort":
|
|
|
265 |
#return OnnxRuntimePredictorSingleton._instance[model_path]
|
266 |
return OnnxRuntimePredictor(**kwargs)
|
267 |
|
|
|
268 |
def get_predictor(**kwargs):
|
269 |
predict_type = kwargs.get("predict_type", "trt")
|
270 |
if predict_type == "ort":
|