fix gradio

.gitignore

@@ -0,0 +1 @@
+.venv

app.py

@@ -12,9 +12,11 @@ torch.set_grad_enabled(False);
 import urllib
 url, filename = ("https://github.com/mateuszbuda/brain-segmentation-pytorch/raw/master/assets/TCGA_CS_4944.png", "TCGA_CS_4944.png")
 try: urllib.URLopener().retrieve(url, filename)
-except: urllib.request.urlretrieve(url, filename)
+except Exception: urllib.request.urlretrieve(url, filename)
 def inference(img):
-    path = img.name
+    print(f">>>>>>>>>>>>> {img}")
+
+    path = img
     slices = [tio.ScalarImage(path).data]
     tensor = torch.cat(slices, dim=-1)
     guessed_affine = np.diag([-1, -1, 9, 1])
@@ -64,4 +66,4 @@ article = "<p style='text-align: center'><a href='https://mateuszbuda.github.io/
 examples = [
             ['TCGA_CS_4944.png']
 ]
-gr.Interface(inference, gr.inputs.Image(label="input image", type='filepath'), gr.outputs.Image(type='plot'), description=description, article=article, title=title, examples=examples, analytics_enabled=False).launch()
+gr.Interface(inference, gr.Image(label="input image", type='filepath'), gr.Plot(), description=description, article=article, title=title, examples=examples, analytics_enabled=False).launch()

flagged/log.csv

@@ -0,0 +1,2 @@
+input image,output,flag,username,timestamp
+flagged\input image\63b444ef562812ef6eb4\TCGA_CS_4944.png,,,,2024-03-21 14:45:02.168756