Refactor image loading in app.py

app.py

@@ -78,7 +78,7 @@ with gr.Blocks(css=css) as demo:
         # image = transform({'image': image})['image']
         # image = torch.from_numpy(image).unsqueeze(0).to(DEVICE)
 
-        image = np.asarray(Image.open(image), dtype=np.float32) / 255.0
+        image = np.asarray(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)