Блог

Я пытаюсь загрузить модель с обнимающего лица, и я скачал модель h5 отсюда: https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english/tree/main

 from flask import Flask, jsonify, request  # import objects from the Flask model
from keras.models import load_model
from transformers import AutoTokenizer, AutoModelForSequenceClassification,TextClassificationPipeline

model = load_model('./tf_model.h5') # trying to load model here
 

И появляется ошибка:

 File "C:DLearningFlaskflask-pp-restmain.py", line 11, in <module>
    model = load_model('./tf_model.h5') File "C:UsersndrezAppDataLocalProgramsPythonPython39libsite-packageskerassavingsave.py",
line 200, in load_model
    return hdf5_format.load_model_from_hdf5(filepath, custom_objects, File
"C:UsersndrezAppDataLocalProgramsPythonPython39libsite-packageskerassavinghdf5_format.py",
line 176, in load_model_from_hdf5
    raise ValueError('No model found in config file.') ValueError: **No model found in config file.**
 

Кто-нибудь знает, как это решить? Если вы знаете, пожалуйста, помогите мне. Я буду следить за этим вопросом и попытаюсь реализовать ответ вашего решения.

Вы можете загрузить версию tensorflow distilbert-base-uncased-finetuned-sst-2-english с помощью класса TFAutoModelForSequenceClassification:

 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = TFAutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
 

Чтобы загрузить указанную вами модель, вот код:

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")