Update app.py

app.py

@@ -17,7 +17,8 @@ class MyModel(torch.nn.Module):
 # 모델 초기화
 num_classes = 6  # 클래스 수에 맞게 설정
 model = MyModel(num_classes)
-model.load_state_dict(torch.load('resnet18_finetuned_2.pth'), strict=False)
+# 모델 가중치 로드
+model.load_state_dict(torch.load('resnet18_finetuned_2.pth', weights_only=True), strict=False)
 model.eval()
 
 # 데이터 변환 정의