Update app.py

app.py

@@ -81,7 +81,7 @@ class BERTClassifier(nn.Module):
         return probabilities  # 각 클래스에 대한 확률 반환
 
 
-model = torch.load('./model_weights_softmax(model).pth')
+model = torch.load('./model_weights_softmax(model).pth', map_location=torch.device('cpu'))
 model.eval()
 
 # 멜론 데이터 불러오기