KeyError в Django при использовании API Huggingface

Я использую Django и хочу использовать Huggingface API с ним. API иногда возвращает мне ошибку, говорящую:

Ошибка ключа в /GenerativeImage2Text

0  

                34. if "generated_text" in output[0]:
                                            ^^^^^^^^

вот мой view.py

def GIT(request):
    output = None
    generated_text = None
    form = ImageForm()  

    if request.method == 'POST':
        form = ImageForm(request.POST, request.FILES)

        if form.is_valid():
            form.save()

            image_name = str(form.cleaned_data['Image'])
            imgloc = os.path.join(
                'media', 'images', image_name)
            while True:
                output = query(imgloc)
                if "generated_text" in output[0]:
                    generated_text = output[0].get('generated_text')
                    break
                else:
                    print(output[0])

        else:
            print("This image is invalid")

    return render(request, "imagecaptioning.html", {"form": form, "output": generated_text})

• я хочу исправить эту проблему и убедиться, что API всегда не возвращает ошибку, такую что.

• находится ли проблема в моем коде или в API?

заранее спасибо

Вернуться на верх