File size: 1,574 Bytes
b3b38b5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# -*- coding: utf-8 -*-
"""demo_gradio_huggingface_spaces.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1vy72Qs-scdzH68lhYbGN11fLzcE9Md1O
"""

#!pip install transformers gradio sentencepiece
#!pip install transformers[sentencepiece]
#!pip install sacremoses
# gradio ๊ทธ๋ผ๋””์˜ค ๊ทธ๋ ˆ์ด๋””์˜ค
# !๋Š” ์ฝ”๋žฉ์—๊ฒŒ ํ•˜๋Š” ์‹œ์Šคํ…œ ๋ช…๋ น์–ด
# ์‹œ์Šคํ…œ ๋ช…๋ น์–ด?๊ฐ€ ๋ญ์ฃ : ํ‚ค์›Œ๋“œ -> ์œ ๋‹‰์Šค ์‰˜ ์Šคํฌ๋ฆฝํŠธ ๋ช…๋ น์–ด
# ! ์—†์ด๋Š” ๊ทธ๋ƒฅ ์ผ๋ฐ˜ ํŒŒ์ด์ฌ ๋ช…๋ น์–ด
# sklearn/pandas/plotly๋Š” ์„ค์น˜ ์•ˆ ํ–ˆ๋Š”๋ฐ?
# ์ฝ”๋žฉ ๋จธ์‹ ์— ๊ธฐ๋ณธ์ ์œผ๋กœ ์„ค์น˜๋˜์–ด์žˆ๋Š”๋ฐ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋“ค์ด ์žˆ์Œ

import gradio as gr # gradio๋ฅผ ๊ฐ€์ ธ์™€์„œ ์“ฐ๋Š”๋ฐ, ๋งค๋ฒˆ gradio ์—ฌ์„ฏ์ž ์น˜๋Š”๊ฒŒ ๊ท€์ฐฎ์œผ๋ฉด
# as gr์ด๋ผ๊ณ  ์ถ”๊ฐ€ํ•ด์„œ ์ด ์ดํ›„๋กœ๋Š” gr์ด๋ผ๊ณ ๋งŒ ์จ๋„ ๋จ
# gradio๊ฐ€ ์›น ์ƒ์—์„œ์˜ ๋ฐ๋ชจ๋ฅผ ๋งŒ๋“ค์–ด์คŒ (UI ํฌํ•จ!)
import transformers # ํŠธ๋žœ์Šคํฌ๋จธ(์‹ ๊ฒฝ๋ง์˜ ์ผ์ข…)
from transformers import pipeline # ํŒŒ์ดํ”„๋ผ์ธ์€ ๋จธ์‹ ๋Ÿฌ๋‹ ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๋Š” ๋ฐ ํ•„์š”ํ•œ
# ์—ฌ๋Ÿฌ ๊ตฌ์„ฑ ์š”์†Œ๋ฅผ ๋ฌถ์–ด์คŒ -> ๊ตฌ์„ฑ ์š”์†Œ๋ž€? ์˜ˆ) ๋ฌธ์ œ์˜ ์ข…๋ฅ˜, ๋ชจ๋ธ์˜ ์ข…๋ฅ˜ ๋“ฑ

pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
# ๋ฌธ์ œ์˜ ์ข…๋ฅ˜๋Š” ๋ฒˆ์—ญ์ด๊ณ , ์šฐ๋ฆฌ๊ฐ€ ์“ธ ๋ชจ๋ธ์€ Hel.... en์ด๋ผ ๋ถˆ๋ฆฌ๋Š” ๋ชจ๋ธ
demo = gr.Interface.from_pipeline(pipe)
# gradio๋ผ๋Š” ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์— ์ •์˜๋˜์–ด์žˆ๋Š” Interface๋ฅผ ๋งŒ๋“ค์ž
# ๋ญ๋กœ๋ถ€ํ„ฐ ๋งŒ๋“œ๋ƒ๋ฉด pipe๋กœ๋ถ€ํ„ฐ
# UI -> User Interface
demo.launch()

# [email protected]