Как успешно развернуть модель трансформера с помощью Gradio в Hugginface Space?

Я хочу развернуть модель трансформатора, классифицирующую изображения, с помощью Gradio в моем пространстве Hugginface. Я создал новый репозиторий и создалapp.pyфайл в браузере в соответствии с руководством на веб-сайте Gradio. Ниже мой код развертывания:

      from transformers import pipeline
import gradio as gr
import timm

def image_classifier(image):
    model = pipeline("image-classification")
    return model(image)

gr.Interface.from_pipeline(model).launch()

Но когда я загружаю файл, я получаю ошибку времени выполнения, в которой говорится:

      Traceback (most recent call last):
  File "app.py", line 1, in <module>
    from transformers import pipeline
ModuleNotFoundError: No module named 'transformers'

Я пытаюсь включитьpip install transformersв сценарии, но я все еще получаю ошибку времени выполнения. Пожалуйста, как я могу успешно развернуть модель трансформера с интерфейсом Gradio в моем пространстве Huggingface? Как всегда, я ценю вашу помощь.

0 ответов

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