Update app.py

app.py

@@ -7,12 +7,13 @@ import torch
 from torch.autograd import Variable
 from torchvision import transforms
 import torch.nn.functional as F
-import gdown
 import warnings
 warnings.filterwarnings("ignore")
 
-os.system("git clone https://github.com/xuebinqin/DIS")
-os.system("mv DIS/IS-Net/* .")
+# Sprawdzenie, czy katalog DIS istnieje przed klonowaniem
+if not os.path.exists("DIS"):
+    os.system("git clone https://github.com/xuebinqin/DIS")
+    os.system("mv DIS/IS-Net/* .")
 
 # project imports
 from data_loader_cache import normalize, im_reader, im_preprocess 
@@ -21,9 +22,10 @@ from models import *
 # Helpers
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-# Download official weights
+# Sprawdzenie, czy katalog z modelami istnieje przed przeniesieniem
 if not os.path.exists("saved_models"):
     os.mkdir("saved_models")
+if not os.path.exists("saved_models/isnet.pth"):
     os.system("mv isnet.pth saved_models/")
 
 class GOSNormalize(object):
@@ -193,22 +195,20 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
     article = gr.Markdown(translations["en"]["article"])
 
     with gr.Row():
-        prompt = gr.Image(type='filepath')
-        result = gr.Image(label="Segmented Image", show_label=False)
-
+        with gr.Column(elem_id="col-container"):
+            gr.Image("logo.png", elem_id="logo-img", show_label=False, show_share_button=False, show_download_button=False)
+            inputs = gr.Image(type='filepath', label="Wybierz obraz")
+            outputs = [gr.Image(label="Wynik (z przezroczystością)"), gr.Image(label="Maska")]
+
+            run_button = gr.Button("Segmentuj", scale=0)
+            
+            run_button.click(fn=inference, inputs=inputs, outputs=outputs)
+            
     language_selector.change(
         fn=change_language,
         inputs=language_selector,
         outputs=[title, description, article],
+        api_name=False,
     )
 
-    gr.Interface(
-        fn=inference,
-        inputs=prompt,
-        outputs=[result],
-        title=title,
-        description=description,
-        article=article,
-        allow_flagging='never',
-        cache_examples=False,
-    ).launch()
+demo.launch()