Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import os
|
|
| 5 |
|
| 6 |
REST_API_KEY = os.environ.get("REST_API_KEY")
|
| 7 |
|
| 8 |
-
def kogpt_api(prompt, max_tokens =
|
| 9 |
r = requests.post(
|
| 10 |
url='https://api.kakaobrain.com/v1/inference/kogpt/generation',
|
| 11 |
headers = {
|
|
@@ -38,10 +38,10 @@ iface = gr.Interface(
|
|
| 38 |
fn=greet,
|
| 39 |
inputs=[
|
| 40 |
gr.Textbox(lines=2, placeholder="Input Prompt Here..."),
|
| 41 |
-
|
| 42 |
-
gr.Slider(0.1, 1.0, step=0.1),
|
| 43 |
-
gr.Slider(0.1, 1.0, step=0.1),
|
| 44 |
-
gr.Slider(1, 16)],
|
| 45 |
outputs=["text"]
|
| 46 |
)
|
| 47 |
|
|
|
|
| 5 |
|
| 6 |
REST_API_KEY = os.environ.get("REST_API_KEY")
|
| 7 |
|
| 8 |
+
def kogpt_api(prompt, max_tokens = 32, 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 = {
|
|
|
|
| 38 |
fn=greet,
|
| 39 |
inputs=[
|
| 40 |
gr.Textbox(lines=2, placeholder="Input Prompt Here..."),
|
| 41 |
+
gr.Number(default=32),
|
| 42 |
+
gr.Slider(0.1, 1.0, step=0.1, default=0.5),
|
| 43 |
+
gr.Slider(0.1, 1.0, step=0.1, default=0.5),
|
| 44 |
+
gr.Slider(1, 16, step=0.1, default=1)],
|
| 45 |
outputs=["text"]
|
| 46 |
)
|
| 47 |
|