Spaces:

ninenox
/

classify_blood

Runtime error

ninenox commited on Apr 15, 2024

Commit

c0f45b2

verified ·

1 Parent(s): 8af4a3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 # โหลด model จากไฟล์ .pt
 model = torch.load('model_blood.pt')
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model.to(device)
 model.eval()

 # โหลด model จากไฟล์ .pt
 model = torch.load('model_blood.pt')
+device = 'cpu' # torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model.to(device)
 model.eval()