Spaces:
Sleeping
Sleeping
File size: 379 Bytes
c92c0ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import os
import sys
migc_path = os.path.dirname(os.path.abspath(__file__))
print(migc_path)
if migc_path not in sys.path:
sys.path.append(migc_path)
from model_bbox.MIGC.inference_single_image import MIGC_Pipe
def load_local_model(model_name, model_type):
if model_name == "MIGC":
pipe = MIGC_Pipe()
else:
raise NotImplementedError
return pipe |