Как успешно развернуть модель трансформера с помощью 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? Как всегда, я ценю вашу помощь.