Update image preprocessing in app.py

app.py

@@ -78,7 +78,8 @@ with gr.Blocks(css=css) as demo:
         # image = transform({'image': image})['image']
         # image = torch.from_numpy(image).unsqueeze(0).to(DEVICE)
 
-        image = torch.from_numpy(image.transpose((2, 0, 1))).to(DEVICE)
+        image = np.asarray(Image.open(image), dtype=np.float32) / 255.0
+        image = torch.from_numpy(image.transpose((2, 0, 1)))
         image = Normalize(mean=[0.485, 0.456, 0.406], std=[
             0.229, 0.224, 0.225])(image)
         image = torch.from_numpy(image).unsqueeze(0).to(DEVICE)