seawolf2357 commited on
Commit
158913f
ยท
verified ยท
1 Parent(s): ef8b4e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,11 +1,12 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
  import requests
 
4
 
5
  # ์ด๋ฏธ์ง€ ์ธ์‹ ํŒŒ์ดํ”„๋ผ์ธ ๋กœ๋“œ
6
  image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
7
 
8
- # API ํ† ํฐ ๋กœ๋“œ
9
  hugging_face_auth_token = os.getenv("HUGGING_FACE_AUTH_TOKEN")
10
 
11
  def get_audiogen(prompt):
@@ -17,6 +18,7 @@ def get_audiogen(prompt):
17
  json={"inputs": prompt, "parameters": {"length": 10}, "options": {"use_cache": False}}
18
  )
19
  result = response.json()
 
20
  return result
21
 
22
  def classify_and_generate_audio(uploaded_image):
 
1
  import gradio as gr
2
  from transformers import pipeline
3
  import requests
4
+ import os # os ๋ชจ๋“ˆ์„ ์ž„ํฌํŠธํ•ฉ๋‹ˆ๋‹ค.
5
 
6
  # ์ด๋ฏธ์ง€ ์ธ์‹ ํŒŒ์ดํ”„๋ผ์ธ ๋กœ๋“œ
7
  image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
8
 
9
+ # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ Hugging Face API ํ† ํฐ ๋กœ๋“œ
10
  hugging_face_auth_token = os.getenv("HUGGING_FACE_AUTH_TOKEN")
11
 
12
  def get_audiogen(prompt):
 
18
  json={"inputs": prompt, "parameters": {"length": 10}, "options": {"use_cache": False}}
19
  )
20
  result = response.json()
21
+ # ์—ฌ๊ธฐ์—์„œ result ์ฒ˜๋ฆฌ ๋กœ์ง์„ ๊ตฌํ˜„ํ•ฉ๋‹ˆ๋‹ค.
22
  return result
23
 
24
  def classify_and_generate_audio(uploaded_image):