성공1 AWS S3 커스텀 모델에 json 뉴스 결과 저장 성공!! 와... 드디어 S3에 올려진 모델에 json 뉴스title을 통과시켜서 어떤 카테고리에 저장될지 예측하여 다시 S3에 업로드 시키는 과정을 성공했다! 처음 클라우드를 써보는거라 너무 어려웠는데 성공해서 너무 기쁘당.. 히히 잘 기록해두고 다음에도 모델 사용할때 써야겠다! import torch from transformers import BertConfig, BertTokenizer, BertForSequenceClassification import json from mymodel import MyModel1 import boto3 def load_model(model_path, number_of_labels, tokenizer_path='bert-base-uncased'): config = BertC.. 2023. 11. 22. 이전 1 다음 728x90
