Upload app.py

app.py

@@ -1,3 +1,5 @@
+import os
+
 import gradio as gr
 from PIL import Image
 
@@ -46,31 +48,31 @@ def generate_ui(key):
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("#### Render")
-                img_bn = gr.Image(f"{img_b}.png")
+                img_bn = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_b}.png"))
             with gr.Column(scale=1):
                 gr.Markdown("#### Background")
-                img_bb = gr.Image(f"{img_b}_background.png")
+                img_bb = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_b}_background.png"))
             with gr.Column(scale=1):
                 gr.Markdown("#### Foreground")
-                img_bf = gr.Image(f"{img_b}_foreground.png")
+                img_bf = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_b}_foreground.png"))
             with gr.Column(scale=1):
                 gr.Markdown("#### Segmentation")
-                img_bs = gr.Image(f"{img_b}_segmentation.png")
+                img_bs = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_b}_segmentation.png"))
 
     gr.Markdown("### After damage")
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("#### Render")
-            img_an = gr.Image(f"{img_a}.png")
+            img_an = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_a}.png"))
         with gr.Column(scale=1):
             gr.Markdown("#### Background")
-            img_ab = gr.Image(f"{img_a}_background.png")
+            img_ab = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_a}_background.png"))
         with gr.Column(scale=1):
             gr.Markdown("#### Foreground")
-            img_af = gr.Image(f"{img_a}_foreground.png")
+            img_af = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_a}_foreground.png"))
         with gr.Column(scale=1):
             gr.Markdown("#### Segmentation")
-            img_as = gr.Image(f"{img_a}_segmentation.png")
+            img_as = gr.Image(os.path.join(os.path.dirname(__file__),f"{img_a}_segmentation.png"))
 
     model = gr.Model3D(model, label="3D model preview")
     return img_bn, img_bb, img_bf, img_bs, img_an, img_ab, img_af, img_as, model