Как развернуть модель трансформера с помощью 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)

Это странно для меня, потому что я точно следовал документации, но получаю сообщение об ошибке. Пожалуйста, ваш совет будет очень ценен.

0 ответов

Другие вопросы по тегам