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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,8 +1,11 @@
1
  import requests
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 = {
@@ -23,7 +26,6 @@ def kogpt_api(REST_API_KEY, prompt, max_tokens = 1, temperature = 1.0, top_p = 1
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),
@@ -35,7 +37,6 @@ def greet(apikey, prompt, max_tokens, temperature, top_p, n):
35
  iface = gr.Interface(
36
  fn=greet,
37
  inputs=[
38
- gr.Textbox(),
39
  gr.Textbox(lines=2, placeholder="Input Prompt Here..."),
40
  "number",
41
  gr.Slider(0.1, 1.0, step=0.1),
 
1
  import requests
2
  import json
3
  import gradio as gr
4
+ import os
5
 
6
+ REST_API_KEY = os.environ.get("REST_API_KEY")
7
+
8
+ def kogpt_api(prompt, max_tokens = 1, temperature = 1.0, top_p = 1.0, n = 1):
9
  r = requests.post(
10
  url='https://api.kakaobrain.com/v1/inference/kogpt/generation',
11
  headers = {
 
26
 
27
  def greet(apikey, prompt, max_tokens, temperature, top_p, n):
28
  response = kogpt_api(
 
29
  prompt = prompt,
30
  max_tokens = int(max_tokens),
31
  temperature = float(temperature),
 
37
  iface = gr.Interface(
38
  fn=greet,
39
  inputs=[
 
40
  gr.Textbox(lines=2, placeholder="Input Prompt Here..."),
41
  "number",
42
  gr.Slider(0.1, 1.0, step=0.1),