bunmei21c commited on
Commit
1bfa953
ยท
verified ยท
1 Parent(s): 2250984

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -6
app.py CHANGED
@@ -1,12 +1,30 @@
1
  import gradio as gr
 
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  demo = gr.Interface(
7
- fn=greet,
8
- inputs="text",
9
- outputs="text"
 
10
  )
11
 
12
- demo.launch() # โ† ์ด ์ค„์ด ์—†์œผ๋ฉด ์ ˆ๋Œ€ /run/predict ํ™œ์„ฑํ™”๋˜์ง€ ์•Š์Œ
 
1
  import gradio as gr
2
+ import openai
3
+ import os
4
 
5
+ # โœ… OpenAI ํ‚ค ์„ค์ • (Hugging Face์—์„œ Secret์œผ๋กœ ์„ค์ •ํ•ด๋„ OK)
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+
8
+ def generate_article(title):
9
+ prompt = f"์ฃผ์ œ: {title}\n\n์ด ์ฃผ์ œ์— ๋Œ€ํ•ด 3~5๋ฌธ๋‹จ ์ •๋„์˜ ์ฝ˜ํ…์ธ  ์ดˆ์•ˆ์„ ์ž‘์„ฑํ•ด์ค˜:"
10
+
11
+ try:
12
+ response = openai.ChatCompletion.create(
13
+ model="gpt-3.5-turbo",
14
+ messages=[
15
+ {"role": "user", "content": prompt}
16
+ ],
17
+ max_tokens=500
18
+ )
19
+ return response['choices'][0]['message']['content']
20
+ except Exception as e:
21
+ return f"[์—๋Ÿฌ] {str(e)}"
22
 
23
  demo = gr.Interface(
24
+ fn=generate_article,
25
+ inputs=gr.Textbox(label="์ฝ˜ํ…์ธ  ์ œ๋ชฉ"),
26
+ outputs=gr.Textbox(label="์ƒ์„ฑ๋œ ์ดˆ์•ˆ", lines=10),
27
+ title="GPT ์ฝ˜ํ…์ธ  ์ดˆ์•ˆ ์ƒ์„ฑ๊ธฐ"
28
  )
29
 
30
+ demo.launch()