Update app.py

app.py

@@ -22,10 +22,7 @@ def load_img (filename):
     return img_tensor
 
 def process_img(image):
-    img = np.array(image)
-    img = img / 255.
-    img = img.astype(np.float32)
-    y = torch.tensor(img).permute(2,0,1).unsqueeze(0).to(device)
+    y = img.to(device)
 
     with torch.no_grad():
         result = model(y)