Update app.py

app.py

@@ -1,3 +1,5 @@
+# インポートと初期設定
+
 import spaces
 import torch
 import gradio as gr
@@ -25,6 +27,10 @@ import os
 from transformers import CLIPImageProcessor
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
 
+
+
+# モデルの初期化
+
 BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
 
 # Initialize both pipelines
@@ -48,6 +54,10 @@ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
     torch_dtype=torch.float16,
 ).to("cuda")
 
+
+
+# 関数の定義
+
 # Function to check NSFW images
 #def check_nsfw_images(images: list[Image.Image]) -> tuple[list[Image.Image], list[bool]]:
 #    if SAFETY_CHECKER_ENABLED:
@@ -134,7 +144,11 @@ def convert_to_base64(pil_image):
         image.save(temp_file.name)
     return temp_file.name
 
+
+
+# 推論関数
 # Inference function
+
 @spaces.GPU
 def inference(
     control_image: Image.Image,
@@ -216,7 +230,11 @@ def inference(
     )
 
     return out_image["images"][0], gr.update(visible=True), gr.update(visible=True), my_seed
-        
+
+
+
+# Gradio UIの構築
+
 with gr.Blocks() as app:
     gr.Markdown(
         '''