Update app.py

app.py

@@ -4,6 +4,13 @@ import torch
 import torchvision.transforms as transforms
 from PIL import Image
 from torchvision import models
+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()
 
 # ข้อความ string ค่าของ classes ที่มี
 targets = ['Negative','Positive']
@@ -34,4 +41,4 @@ demo = gr.Interface(fn=classify_image,
              outputs=gr.Label(num_top_classes=2),
              examples=["examples/negative.jpeg", "examples/positive.jpeg"]) # ภาพตัวอย่างมาจาก folder examples
 
-demo.launch(share=True, debug=True)
+demo.launch(share=True, debug=True)