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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -1,21 +1,20 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # ๋น„์ „ ์ธ์‹ ํŒŒ์ดํ”„๋ผ์ธ ์ดˆ๊ธฐํ™”
5
- vision_pipeline = pipeline(task="image-classification")
6
 
7
  def process_video(video):
8
- # ๋น„๋””์˜ค ํ”„๋ ˆ์ž„์„ ์ฒ˜๋ฆฌํ•˜๊ณ  ๋น„์ „ ์ธ์‹ ๋ชจ๋ธ์„ ์ ์šฉํ•˜๋Š” ํ•จ์ˆ˜
9
- # ์—ฌ๊ธฐ์„œ๋Š” ์˜ˆ์‹œ๋ฅผ ๋‹จ์ˆœํ™”ํ•˜๊ธฐ ์œ„ํ•ด ๋น„๋””์˜ค ํŒŒ์ผ์˜ ์ฒซ ํ”„๋ ˆ์ž„๋งŒ ์ฒ˜๋ฆฌํ•ฉ๋‹ˆ๋‹ค.
10
- # ์‹ค์ œ ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜์—์„œ๋Š” ๋น„๋””์˜ค์˜ ๋ชจ๋“  ํ”„๋ ˆ์ž„ ๋˜๋Š” ํŠน์ • ์ธํ„ฐ๋ฒŒ๋กœ ํ”„๋ ˆ์ž„์„ ์ฒ˜๋ฆฌํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
11
  results = vision_pipeline(video)
12
  labels = [result['label'] for result in results]
13
  return ", ".join(labels)
14
 
15
- # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜
16
  iface = gr.Interface(
17
  fn=process_video,
18
- inputs=gr.Video(source="webcam", streaming=True),
19
  outputs="text",
20
  title="์‹ค์‹œ๊ฐ„ ์›น์บ  ๋ชจ๋‹ˆํ„ฐ๋ง๊ณผ ๋น„์ „ ์ธ์‹",
21
  description="์›น์บ ์„ ํ†ตํ•ด ์‹ค์‹œ๊ฐ„์œผ๋กœ ํ™”๋ฉด ๋ณ€ํ™”๋ฅผ ๊ฐ์ง€ํ•˜๊ณ , ๋น„์ „ ์ธ์‹ ๋ชจ๋ธ์„ ์ ์šฉํ•˜์—ฌ ํ…์ŠคํŠธ๋กœ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค."
 
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="์›น์บ ์„ ํ†ตํ•ด ์‹ค์‹œ๊ฐ„์œผ๋กœ ํ™”๋ฉด ๋ณ€ํ™”๋ฅผ ๊ฐ์ง€ํ•˜๊ณ , ๋น„์ „ ์ธ์‹ ๋ชจ๋ธ์„ ์ ์šฉํ•˜์—ฌ ํ…์ŠคํŠธ๋กœ ์ถœ๋ ฅํ•ฉ๋‹ˆ๋‹ค."