1

app.py

@@ -24,9 +24,18 @@ pipe = pipeline(
 def chat_fn(prompt):
     try:
         output = pipe(prompt)[0]["generated_text"]
-        return [output]
+        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")
 
-demo = gr.Interface(fn=chat_fn, inputs="text", outputs="text")
+    btn.click(fn=chat_fn, inputs=inp, outputs=out)
+
+    # API endpoint 등록
+    demo.api(chat_fn, inputs=gr.Textbox(label="prompt"), outputs=gr.Textbox(), name="predict")
+    
 demo.launch(share=False)