2

app.py

@@ -27,15 +27,28 @@ def chat_fn(prompt):
         return output
     except Exception as e:
         return [f"Error: {str(e)}"]
+
+
 with gr.Blocks() as demo:
-    gr.Markdown("### 🐯 KoAlpaca Chatbot")
-    inp = gr.Textbox(label="Input")
-    out = gr.Textbox(label="Output")
-    btn = gr.Button("Submit")
+    with gr.Row():
+        input_box = gr.Textbox(label="Prompt", lines=2)
+    with gr.Row():
+        output_box = gr.Textbox(label="Response")
+
+    btn = gr.Button("Generate")
+    btn.click(chat_fn, inputs=input_box, outputs=output_box)
+
+    # ✅ Hugging Face Spaces의 API 요청용 endpoint 정의
+    gr.Examples(
+        examples=["안녕?", "한국에 대해 말해줘"],
+        inputs=input_box
+    )
+
+    demo.load(chat_fn, inputs=input_box, outputs=output_box)
 
-    btn.click(fn=chat_fn, inputs=inp, outputs=out)
+# ✅ API endpoint로 사용할 Interface 객체 등록
+api_demo = gr.Interface(fn=chat_fn, inputs="text", outputs="text", name="predict")
 
-    # API endpoint 등록
-    demo.api(chat_fn, inputs=gr.Textbox(label="prompt"), outputs=gr.Textbox(), name="predict")
-    
-demo.launch(share=False)
+if __name__ == "__main__":
+    demo.queue()
+    api_demo.launch(share=False)