Update gradio_ui.py

gradio_ui.py

@@ -287,8 +287,14 @@ def get_image_paths(folder_path):
       image_paths.append([os.path.join(folder_path, filename)])
   return image_paths
 
-# Create the Gradio interface
+
 def create_interface():
+    css = """
+    footer {
+        visibility: hidden;
+    }
+    """
+    
     controlnet_model_dict = {
        "sdxl-light-caption-30000": "sdxl_light_caption_output/checkpoint-30000/controlnet",
        "sdxl-light-custom-caption-30000": "sdxl_light_custom_caption_output/checkpoint-30000/controlnet",
@@ -320,11 +326,6 @@ def create_interface():
         ],
         examples=images,
         additional_inputs=[
-            # gr.Radio(choices=["Original", "Square"], value="Original", 
-            #          label="Output resolution"),
-            # gr.Slider(minimum=128, maximum=512, value=256, step=128,
-            #           label="Height & Width", 
-            #           info='Only effect if select "Square" output resolution'),
             gr.Slider(0, 1000, 123, label="Seed"),
             gr.Radio(choices=[1, 2, 4, 8],
                      value=8,
@@ -364,24 +365,12 @@ def create_interface():
         ],
         title="ColorRevive",
         description="Restore a black-and-white photo to color",
+        theme="Yntec/HaleyCH_Theme_Orange",
+        css=css,
         cache_examples=False
     )
     return interface
 
-def create_interface():
-    css = """
-    footer {
-        visibility: hidden;
-    }
-    """
-    
-    # 테마와 CSS를 인터페이스 생성 시 설정
-    interface = gr.Interface(
-        # ... 기존 파라미터들 ...
-        theme="Yntec/HaleyCH_Theme_Orange",
-        css=css
-    )
-    return interface
 
 def main():
     # launch()에서는 테마 설정을 제거