test

app.py

@@ -6,7 +6,7 @@ import numpy as np
 def pil_to_torch(image):
     image = np.array(image)
     image = image.transpose((2, 0, 1))
-    image = torch.tensor(image)
+    image = torch.tensor(image).float()
     return image.unsqueeze(0)
 
 def torch_to_pil(image):
@@ -23,10 +23,10 @@ def image_mod(image):
     return torch_to_pil(noisy), torch_to_pil(estimated)
 
 
-input_image = gr.components.Image(label='Input Image')
-output_images = gr.components.Image(label='Denoised Image')
-noise_image = gr.components.Image(label='Noisy Image')
-input_image_output = gr.components.Image(label='Input Image')
+input_image = gr.Image(label='Input Image')
+output_images = gr.Image(label='Denoised Image')
+noise_image = gr.Image(label='Noisy Image')
+input_image_output = gr.Image(label='Input Image')
 
 
 demo = gr.Interface(