replace tabs

app.py

@@ -22,6 +22,7 @@ def clean(text_output):
     else:
         pass
 
+
 def clean_by_name(text_output):
     text_output, dir_name = text_output.split("+")
     if text_output=="complete":
@@ -90,9 +91,13 @@ def from_zip(inputs):
     return "objects.zip", complete()
 
 
-def from_image_files(images):
-    dir_name = functions.get_random_name()
-    os.makedirs(dir_name, exist_ok=True)
+def from_image_files(images, text_class_name):
+
+    if not text_class_name=="":
+        dir_name = text_class_name
+    else:
+        dir_name = functions.get_random_name()
+        os.makedirs(dir_name, exist_ok=True)
 
     for image in images:
         image_name = image.name
@@ -123,6 +128,51 @@ if __name__=="__main__":
        
     with gr.Blocks() as demo:
 
+        
+        with gr.Tab("Images"):
+            gr.Markdown(
+            """
+            <center><h1>Image Matting using U<sup>2</sup>-Net</h1></center>
+            """
+            )        
+            with gr.Row():
+                gr.Markdown(
+                    """
+                    ### Input Image Files
+                    <img src='file/assets/input_images.png' width="90%" height="90%">
+                    """
+                )        
+                gr.Markdown(
+                    """
+                    ### Output Zip File
+                    <img src='file/assets/output_zip.png' width="85%" height="85%">
+                    """
+                )        
+            
+            with gr.Row():
+                with gr.Column():
+                    text_class_name = gr.Textbox(label="Class Name", value="", placeholder="cat")
+                    image_input = gr.File(file_count="multiple")
+                image_output = gr.File()
+                text_output = gr.Textbox(visible=False)
+            
+            btn = gr.Button("Run!")
+            
+            btn.click(
+                fn=from_image_files,
+                inputs=[image_input, text_class_name],
+                outputs=[image_output, text_output]
+                )
+            text_output.change(
+                fn=clean_by_name,
+                inputs=text_output,
+                outputs=text_output
+
+            )
+
+
+
+
         with gr.Tab("Zip"):
             gr.Markdown(
             """
@@ -198,47 +248,6 @@ if __name__=="__main__":
                 )
         
 
-        
-        with gr.Tab("Images"):
-            gr.Markdown(
-            """
-            <center><h1>Image Matting using U<sup>2</sup>-Net</h1></center>
-            """
-            )        
-            with gr.Row():
-                gr.Markdown(
-                    """
-                    ### Input Image Files
-                    <img src='file/assets/input_zip.png' width="85%" height="85%">
-                    """
-                )        
-                gr.Markdown(
-                    """
-                    ### Output Zip File
-                    <img src='file/assets/output_zip.png' width="85%" height="85%">
-                    """
-                )        
-            
-            with gr.Row():
-                image_input = gr.File(file_count="multiple")
-                image_output = gr.File()
-                text_output = gr.Textbox(visible=False)
-            
-            btn = gr.Button("Run!")
-            
-            btn.click(
-                fn=from_image_files,
-                inputs=image_input,
-                outputs=[image_output, text_output]
-                )
-            text_output.change(
-                fn=clean_by_name,
-                inputs=text_output,
-                outputs=text_output
-
-            )
-
-
 
 
         gr.Markdown(