Update app.py

app.py

@@ -1,23 +1,22 @@
 import gradio as gr
 from transformers import pipeline
 import requests
-import os
-
-# 환경변수에서 Hugging Face API 토큰을 로드
-hugging_face_auth_token = os.getenv("HUGGING_FACE_AUTH_TOKEN")
 
 # 이미지 인식 파이프라인 로드
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
+# API 토큰 로드
+hugging_face_auth_token = os.getenv("HUGGING_FACE_AUTH_TOKEN")
+
 def get_audiogen(prompt):
     # 오디오 생성 모델 API 호출
+    headers = {"Authorization": f"Bearer {hugging_face_auth_token}"}
     response = requests.post(
         "https://api-inference.huggingface.co/models/fffiloni/audiogen",
-        headers={"Authorization": f"Bearer {hugging_face_auth_token}"},  # 수정된 부분
+        headers=headers,
         json={"inputs": prompt, "parameters": {"length": 10}, "options": {"use_cache": False}}
     )
     result = response.json()
-    # 생성된 오디오 파일의 URL을 반환하거나, 오디오 데이터 자체를 반환
     return result
 
 def classify_and_generate_audio(uploaded_image):