seawolf2357 commited on
Commit
3fb0324
ยท
verified ยท
1 Parent(s): bcc0c38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -1,25 +1,25 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
 
4
- # ๋น„์ „ ์ธ์‹ ํŒŒ์ดํ”„๋ผ์ธ์„ ๋ช…์‹œ์ ์œผ๋กœ ๋ชจ๋ธ ์ด๋ฆ„๊ณผ ํ•จ๊ป˜ ์ดˆ๊ธฐํ™”
5
- vision_pipeline = pipeline(task="image-classification", model="google/vit-base-patch16-224")
 
 
 
 
 
 
 
 
6
 
7
- def process_video(video):
8
- # ๋น„๋””์˜ค ํ”„๋ ˆ์ž„์„ ์ฒ˜๋ฆฌํ•˜๋Š” ํ•จ์ˆ˜
9
- # ์—ฌ๊ธฐ์„œ๋Š” ๋‹จ์ˆœํ™”๋ฅผ ์œ„ํ•ด ๋น„๋””์˜ค ํŒŒ์ผ์˜ ์ฒซ ํ”„๋ ˆ์ž„๋งŒ ์ฒ˜๋ฆฌํ•ฉ๋‹ˆ๋‹ค.
10
- results = vision_pipeline(video)
11
- labels = [result['label'] for result in results]
12
- return ", ".join(labels)
13
-
14
- # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜, `source` ์ธ์ž๋ฅผ ์ œ๊ฑฐ
15
- iface = gr.Interface(
16
- fn=process_video,
17
- inputs=gr.Video(streaming=True),
18
  outputs="text",
19
- title="์‹ค์‹œ๊ฐ„ ์›น์บ  ๋ชจ๋‹ˆํ„ฐ๋ง๊ณผ ๋น„์ „ ์ธ์‹",
20
- description="์›น์บ ์„ ํ†ตํ•ด ์‹ค์‹œ๊ฐ„์œผ๋กœ ํ™”๋ฉด ๋ณ€ํ™”๋ฅผ ๊ฐ์ง€ํ•˜๊ณ , ๋น„์ „ ์ธ์‹ ๋ชจ๋ธ์„ ์ ์šฉํ•˜์—ฌ ํ…์ŠคํŠธ๋กœ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค."
21
  )
22
 
23
- # Gradio ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์‹คํ–‰
24
- if __name__ == "__main__":
25
- iface.launch()
 
1
  import gradio as gr
2
+ from googletrans import Translator
3
 
4
+ def translate_to_english_keywords(text):
5
+ # Translator ๊ฐ์ฒด ์ƒ์„ฑ
6
+ translator = Translator()
7
+
8
+ # ์ž…๋ ฅ๋ฐ›์€ ํ•œ๊ธ€ ๋ฌธ์žฅ์„ ์˜์–ด๋กœ ๋ฒˆ์—ญ
9
+ translated_text = translator.translate(text, src='ko', dest='en')
10
+
11
+ # ๋ฒˆ์—ญ๋œ ์˜์–ด ๋ฌธ์žฅ์—์„œ ํ‚ค์›Œ๋“œ ์ถ”์ถœ(์—ฌ๊ธฐ์„œ๋Š” ๊ฐ„๋‹จํžˆ ์ „์ฒด ๋ฒˆ์—ญ๋œ ๋ฌธ์žฅ์„ ๋ฐ˜ํ™˜)
12
+ # ์‹ค์ œ ํ™œ์šฉ ์‹œ์—๋Š” ํ‚ค์›Œ๋“œ ์ถ”์ถœ ๋กœ์ง์„ ์ถ”๊ฐ€ํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
13
+ return translated_text.text
14
 
15
+ # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜
16
+ interface = gr.Interface(
17
+ fn=translate_to_english_keywords,
18
+ inputs=gr.inputs.Textbox(lines=2, placeholder="ํ•œ๊ธ€ ๋ฌธ์žฅ์„ ์ž…๋ ฅํ•˜์„ธ์š”..."),
 
 
 
 
 
 
 
19
  outputs="text",
20
+ title="ํ•œ๊ธ€ ๋ฌธ์žฅ์„ ์˜์–ด ํ‚ค์›Œ๋“œ๋กœ ๋ฒˆ์—ญ",
21
+ description="ํ•œ๊ธ€ ๋ฌธ์žฅ์„ ์ž…๋ ฅํ•˜๋ฉด, ๊ทธ ์˜๋ฏธ๊ฐ€ ํฌํ•จ๋œ ๊ฐ„๋žตํ•˜๊ณ  ํ•จ์ถ•์ ์ธ ์˜์–ด๋กœ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค."
22
  )
23
 
24
+ # ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์‹คํ–‰
25
+ interface.launch()