arxivgpt kim commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,18 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
gr.Textbox(label="๋์ ์ฒญ์ค"),
|
| 11 |
-
gr.Dropdown(choices=["๊ณต์์ ์ธ", "๋น๊ณต์์ ์ธ", "์ค๋๋ ฅ ์๋"], label="ํค"),
|
| 12 |
-
gr.Number(label="๋จ์ด ์"),
|
| 13 |
-
gr.Textbox(label="์คํ์ผ"),
|
| 14 |
-
gr.Dropdown(choices=["๋ธ๋ก๊ทธ ํฌ์คํธ", "๋ณด๋ ์๋ฃ", "๊ธฐ์ฌ"], label="์ฝํ
์ธ ํ์")],
|
| 15 |
-
outputs="text",
|
| 16 |
-
title="Write For Me - ์ฝํ
์ธ ์์ฑ๊ธฐ",
|
| 17 |
-
description="์ฌ์ฉ์์ ์๊ตฌ์ฌํญ์ ๋ง์ถฐ ์ฝํ
์ธ ๋ฅผ ์์ฑํฉ๋๋ค.") as interface:
|
| 18 |
-
interface.launch()
|
|
|
|
| 1 |
+
# ํ์ํ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ค์นํฉ๋๋ค
|
| 2 |
+
# pip install gradio transformers
|
| 3 |
+
|
| 4 |
+
from transformers import pipeline
|
| 5 |
import gradio as gr
|
| 6 |
|
| 7 |
+
# GPT ๋ชจ๋ธ์ ์ฌ์ฉํ๋ ํ์ดํ๋ผ์ธ์ ์ด๊ธฐํํฉ๋๋ค. ๋ชจ๋ธ ์๋ณ์๋ ์ฌ์ฉํ๊ณ ์ ํ๋ GPT ๋ชจ๋ธ์ ๋ฐ๋ผ ๋ฌ๋ผ์ง๋๋ค.
|
| 8 |
+
# ์: 'gpt-2', 'EleutherAI/gpt-neo-2.7B' ๋ฑ. GPT-4๋ฅผ ์ฌ์ฉํ ๊ฒฝ์ฐ ํด๋น ๋ชจ๋ธ์ ์ ํํ ์๋ณ์๋ฅผ ์ฌ์ฉํด์ผ ํฉ๋๋ค.
|
| 9 |
+
text_generator = pipeline('text-generation', model='gpt-4-model-identifier')
|
| 10 |
+
|
| 11 |
+
def generate_text(prompt, max_length=50, temperature=0.7):
|
| 12 |
+
# ํจ์๋ ์ฃผ์ด์ง ํ๋กฌํํธ๋ฅผ ๋ฐํ์ผ๋ก ํ
์คํธ๋ฅผ ์์ฑํฉ๋๋ค.
|
| 13 |
+
generated_texts = text_generator(prompt, max_length=max_length, temperature=temperature, return_full_text=False)
|
| 14 |
+
return generated_texts[0]['generated_text']
|
| 15 |
+
|
| 16 |
+
# Gradio ์ธํฐํ์ด์ค๋ฅผ ๊ตฌ์ฑํฉ๋๋ค.
|
| 17 |
+
interface = gr.Interface(
|
| 18 |
+
fn=generate_text, # ํธ์ถํ ํจ์
|
| 19 |
+
inputs=[ # ์ฌ์ฉ์๋ก๋ถํฐ ์
๋ ฅ๋ฐ์ ์์๋ค
|
| 20 |
+
gr.inputs.Textbox(lines=5, label="Input Text"),
|
| 21 |
+
gr.inputs.Slider(minimum=10, maximum=100, default=50, label="Maximum Length"),
|
| 22 |
+
gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0.7, label="Temperature")
|
| 23 |
+
],
|
| 24 |
+
outputs=gr.outputs.Textbox(label="Generated Text"), # ๊ฒฐ๊ณผ๋ฅผ ํ์ํ ์์
|
| 25 |
+
title="GPT-4 Text Generator", # ์ธํฐํ์ด์ค์ ์ ๋ชฉ
|
| 26 |
+
description="Enter text to see how GPT-4 can expand, summarize, or rewrite it." # ์ธํฐํ์ด์ค์ ์ค๋ช
|
| 27 |
+
)
|
| 28 |
|
| 29 |
+
# ์ธํฐํ์ด์ค๋ฅผ ์คํํฉ๋๋ค.
|
| 30 |
+
interface.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|