Spaces:
Runtime error
Runtime error
from mmdet.apis import init_detector, inference_detector | |
import mmcv | |
import gradio as gr | |
config_file = 'config/fashionformer_fashionpedia.py' | |
checkpoint_file = 'checkpoints/fashionformer.pth' | |
model = init_detector(config_file, checkpoint_file, device='cpu') | |
def predict(img): | |
result = inference_detector(model, img) | |
model.show_result(img, result, out_file='result.jpg', show=False) | |
return 'result.jpg' | |
gr.Interface(fn=predict, inputs=gr.Image(type="filepath"), outputs="image").launch() |