Create app.py

app.py

@@ -0,0 +1,46 @@
+import requests
+import json
+import gradio as gr
+
+def kogpt_api(prompt, max_tokens = 1, temperature = 1.0, top_p = 1.0, n = 1):
+    r = requests.post(
+        url='https://api.kakaobrain.com/v1/inference/kogpt/generation',
+        headers = {
+            'Authorization': 'KakaoAK ' + REST_API_KEY,
+            'Content-Type': 'application/json'
+        },
+        json = {
+            'prompt': prompt,
+            'max_tokens': max_tokens,
+            'temperature': temperature,
+            'top_p': top_p,
+            'n': n
+        }
+    )
+    # 응답 JSON 형식으로 변환
+    response = json.loads(r.content)
+    return response
+    
+def greet(apikey, prompt, max_tokens, temperature, top_p, n):
+    response = kogpt_api(
+        prompt = prompt,
+        max_tokens = int(max_tokens),
+        temperature = float(temperature),
+        top_p = float(top_p),
+        n = int(n)
+    )
+    return response['generations'][0]['text']
+
+iface = gr.Interface(
+    fn=greet,
+    inputs=[
+        gr.Textbox(),
+        gr.Textbox(lines=2, placeholder="Input Prompt Here..."),
+        "number",
+        gr.Slider(0.1, 1.0, step=0.1),
+        gr.Slider(0.1, 1.0, step=0.1),
+        gr.Slider(1, 16)],
+    outputs=["text"]
+    )
+
+iface.launch()