test

app.py

@@ -1,21 +1,21 @@
 import gradio as gr
 import os
+import deepinv as dinv
+import torch
 
 
 def image_mod(image):
-    return image.rotate(45)
+    denoiser = dinv.models.BM3D()
+    image = torch.tensor(image).unsqueeze(0)
+    image = denoiser(image)
+    image = image.squeeze(0).cpu().detach().numpy()
+    return image
 
 demo = gr.Interface(
     image_mod,
     gr.Image(type="pil"),
     "image",
     flagging_options=["blurry", "incorrect", "other"],
-    examples=[
-        os.path.join(os.path.dirname(__file__), "images/cheetah1.jpg"),
-        os.path.join(os.path.dirname(__file__), "images/lion.jpg"),
-        os.path.join(os.path.dirname(__file__), "images/logo.png"),
-        os.path.join(os.path.dirname(__file__), "images/tower.jpg"),
-    ],
 )
 
 demo.launch()