temion commited on
Commit
1dfcb9d
·
1 Parent(s): ba9865c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import requests
2
  import json
3
  import gradio as gr
4
 
5
- def kogpt_api(prompt, max_tokens = 1, temperature = 1.0, top_p = 1.0, n = 1):
6
  r = requests.post(
7
  url='https://api.kakaobrain.com/v1/inference/kogpt/generation',
8
  headers = {
@@ -17,12 +17,13 @@ def kogpt_api(prompt, max_tokens = 1, temperature = 1.0, top_p = 1.0, n = 1):
17
  'n': n
18
  }
19
  )
20
- # 응답 JSON 형식으로 변환
21
  response = json.loads(r.content)
22
  return response
23
-
24
  def greet(apikey, prompt, max_tokens, temperature, top_p, n):
25
  response = kogpt_api(
 
26
  prompt = prompt,
27
  max_tokens = int(max_tokens),
28
  temperature = float(temperature),
 
2
  import json
3
  import gradio as gr
4
 
5
+ def kogpt_api(REST_API_KEY, prompt, max_tokens = 1, temperature = 1.0, top_p = 1.0, n = 1):
6
  r = requests.post(
7
  url='https://api.kakaobrain.com/v1/inference/kogpt/generation',
8
  headers = {
 
17
  'n': n
18
  }
19
  )
20
+
21
  response = json.loads(r.content)
22
  return response
23
+
24
  def greet(apikey, prompt, max_tokens, temperature, top_p, n):
25
  response = kogpt_api(
26
+ REST_API_KEY=apikey,
27
  prompt = prompt,
28
  max_tokens = int(max_tokens),
29
  temperature = float(temperature),