Update app.py

app.py

@@ -57,12 +57,39 @@ if __name__ == "__main__":
         images = np.clip(images / 2 + 0.5, 0, 1)
         return im.fromarray((images[0] * 255).astype(np.uint8))
 
-    demo = gr.Interface(
-        generate_image,
-        inputs=[gr.Textbox(placeholder="몽타주를 설명하는 텍스트를 입력하세요.", label="Text Condition"), 
-                gr.Slider(0, 10, value=2, label="Guidance Scale", info="이 숫자가 크면 클수록 입력 텍스트를 더 강하게 이용하여 몽타주를 생성합니다.")],
-        outputs=["image"],
-    )
+    with gr.Blocks() as demo:
+        gr.Markdown(
+            """
+            # KoFace AI - 한국인 몽타주 생성기
+            * 몽타주를 설명하는 텍스트를 입력하고 '생성하기' 버튼을 눌러주세요.
+            
+            * **참고**: 현재 무료 티어를 이용하고 있어, **생성에 약 10분 정도의 시간**이 걸릴 수 있습니다. 
+    
+            * 이 AI는 Latent Diffusion Model을 직접 구현하여 만들었습니다. 자세한 코드나 실험 결과는 GitHub 저장소를 참고해 주세요.
+    
+            🔗 [GitHub 저장소 바로가기](https://github.com/Won-Seong/simple-latent-diffusion-model)
+    
+            
+            """
+        )
+    
+        with gr.Row():
+            with gr.Column():
+                text_input = gr.Textbox(
+                    placeholder="몽타주를 설명하는 텍스트를 입력하세요.", 
+                    label="Text Condition"
+                )
+                guidance_slider = gr.Slider(
+                    0, 10, value=2, 
+                    label="Guidance Scale", 
+                    info="이 숫자가 크면 클수록 입력 텍스트를 더 강하게 이용하여 몽타주를 생성합니다."
+                )
+                submit_btn = gr.Button("생성하기")
+    
+            with gr.Column():
+                image_output = gr.Image(label="생성된 몽타주")
+    
+        submit_btn.click(fn=greet, inputs=[text_input, guidance_slider], outputs=image_output)
     
     demo.launch()