Update app.py

app.py

@@ -5,17 +5,21 @@ from gradio_client import Client  # 가정: gradio_client 라이브러리가 사
 # 이미지 인식 파이프라인 로드
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
-def generate_music(prompt):
+def generate_music(image_classification_result):
      # 기본 프롬프트 설정
      base_prompt = "The rnb beat of 85BPM drums. playing Violin."
-    # 사물 인식 결과를 기본 프롬프트에 추가
-    full_prompt = f"{base_prompt} for a {object_detected}"
     # 부정적인 프롬프트 추가
     negative_prompt = "Low quality."
+
+    # 이미지 분류 결과를 포함한 전체 프롬프트 생성
+    full_prompt = f"{base_prompt} Inspired by: {image_classification_result}."
+
     
     # 음악 생성 API 호출
     client = Client("https://haoheliu-audioldm2-text2audio-text2music.hf.space/")
     result = client.predict(
+        prompt=full_prompt,  # 기본 프롬프트
+        negative_prompt=negative_prompt,  # 부정적인 프롬프트
         prompt,
         prompt,  # 음성 생성에 사용될 프롬프트
         5,  # 음악의 길이 (초)