Как развернуть модель трансформера с помощью Gradio
Я хочу развернуть модель обнаружения объектов-трансформеров Huggingface с помощью Gradio. Я точно следовал документации Gradio для моделей трансформаторов, но каждый раз, когда пытаюсь запустить приложение, я получаю следующее сообщение об ошибке:
ValueError Traceback (most recent call last)
<ipython-input-21-ae95d9ecbd73> in <module>()
----> 1 gr.Interface.from_pipeline(obj_detect).launch(share = True)
1 frames
/usr/local/lib/python3.7/dist-packages/gradio/external.py in load_from_pipeline(pipeline)
575 }
576 else:
--> 577 raise ValueError("Unsupported pipeline type: {}".format(type(pipeline)))
578
579 # define the function that will be called by the Interface
ValueError: Unsupported pipeline type: <class 'transformers.pipelines.object_detection.ObjectDetectionPipeline'>
Это мой код:
Import timm
Import gradio as gr
from transformers import pipeline
obj_detect = pipeline("object-detection")
gr.Interface.from_pipeline(obj_detect).launch(share = True)
И когда я фактически определяю функцию и запускаю приложение, приложение запускается, но возвращает ошибку, когда я пытаюсь сделать прогноз. Вот моя функция-оболочка:
def Object_detector(image):
obj_detect = pipeline("object-detection")
detect = obj_detect(image)
return detect
app = gr.Interface(Object_detector,inputs = "image",outputs = "label")
app.launch(share = True)
Это странно для меня, потому что я точно следовал документации, но получаю сообщение об ошибке. Пожалуйста, ваш совет будет очень ценен.