Update app.py

app.py

@@ -7,9 +7,9 @@ from torchvision import models
 import torch
 
 # โหลด model จากไฟล์ .pt
-model = torch.load('model_blood.pt')
-device = 'cpu' # torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model.to(device)
+model = torch.load('model_blood.pt',map_location=torch.device('cpu'))
+# device = 'cpu' # torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# model.to(device)
 model.eval()
 
 # ข้อความ string ค่าของ classes ที่มี