jason9693 commited on
Commit
8b592ae
β€’
1 Parent(s): 54cedc4
Files changed (1) hide show
  1. app.py +22 -0
app.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ theme = "darkgrass"
5
+ title = "GPT-NeoX(Korean) Demo"
6
+ model_name = "EleutherAI/gpt-neox-ko-1.3b"
7
+
8
+ description = "GPT-NeoX ν•œκ΅­μ–΄ λͺ¨λΈμ„ μ‹œμ—°ν•˜λŠ” 데λͺ¨νŽ˜μ΄μ§€ μž…λ‹ˆλ‹€."
9
+ # article = "<p style='text-align: center'><a href='https://github.com/kingoflolz/mesh-transformer-jax' target='_blank'>GPT-J-6B: A 6 Billion Parameter Autoregressive Language Model</a></p>"
10
+ examples = [
11
+ ["μΈκ°„μ²˜λŸΌ μƒκ°ν•˜κ³ , ν–‰λ™ν•˜λŠ” '지λŠ₯'을 톡해"],
12
+ ["λŒ€ν•œλ―Όκ΅­μ˜ 2040λ…„ 전망은 "],
13
+ ["2040λ…„ 미ꡭ은, "]
14
+ ]
15
+
16
+ generator = pipeline('text-generation', model=model_name, use_auth_token=os.environ['TOKEN'])
17
+ gr.Interface.from_pipeline(generator).launch()
18
+
19
+
20
+
21
+
22
+ # print(generated) # print: μΈκ°„μ²˜λŸΌ μƒκ°ν•˜κ³ , ν–‰λ™ν•˜λŠ” '지λŠ₯'을 톡해 인λ₯˜κ°€ μ΄μ œκΉŒμ§€ 풀지 λͺ»ν–ˆλ˜ 문제의 해닡을 찾을 수 μžˆμ„ 것이닀. κ³Όν•™κΈ°μˆ μ΄ κ³ λ„λ‘œ λ°œλ‹¬ν•œ 21μ„ΈκΈ°λ₯Ό μ‚΄μ•„κ°ˆ 우리 μ•„μ΄λ“€μ—κ²Œ κ°€μž₯ ν•„μš”ν•œ 것은 사고λ ₯ ν›ˆλ ¨μ΄λ‹€. 사고λ ₯ ν›ˆλ ¨μ„ 톡해, 세상