Я пытаюсь создать модель машинного обучения и использовать платформу django rest для ее подключения. Моя структура папок выглядит следующим образом
server
|
+-- chat
| |
| +-- DistilBertModel
| | |
| | +-- OurModel
| | | |
| | | +-- BERT_model.py
| | | +-- chatbot.py
| | | +-- chatbot_predict.py
| | | +-- model.sav
| | +-- data
| | | |
| | | +-- test_chatbot.csv
| | | +-- answers.json
| +-- migrations
| +-- __init__.py
| +-- admin.py
| +-- apps.py
| +-- models.py
| +-- serializers.py
| +-- tests.py
| +-- urls.py
| +-- views.py
+-- core
|
+-- templates
|
+-- db.sqlite3
|
+-- manage.py
|
+-- requirements.txt
В chatbot_predict.py файл Я пытаюсь загрузить модель и получить прогноз, используя эту модель. Код выглядит так
import joblib
loaded_model = joblib.load ('model.sav')
def get_prediction (message, loaded_model = loaded_model):
return prediction
Когда я запускаю этот файл с помощью python chatbot_predict.py с помощью командной строки это работает нормально.
Однако, когда я импортирую эту функцию get_prediction в server/chat/views.py и запустите python manage.py runserver выдает ошибку
Что я делаю не так, чтобы получить этот файлообменник ?