Update app.py

app.py

@@ -6,27 +6,19 @@ from gradio_client import Client  # 가정: gradio_client 라이브러리가 사
 image_model = pipeline("image-classification", model="google/vit-base-patch16-224")
 
 def generate_music(image_classification_result):
-    # 기본 프롬프트 설정
-    base_prompt = "The rnb beat of 85BPM drums. playing Violin."
-    # 부정적인 프롬프트 추가
-    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,
-    duration=5,  # 음악의 길이 (초), 키워드 인자로 변경
-    guidance_scale=0,  # 가이던스 스케일, 키워드 인자로 변경
-    seed=5,  # 시드 값, 키워드 인자로 변경
-    num_waveforms=1,  # 생성할 waveform의 수, 키워드 인자로 변경
+    "The rnb beat of 85BPM drums. playing Violin."
+    "Low quality."
+    5,  # 음악의 길이 (초), 키워드 인자로 변경
+    0,  # 가이던스 스케일, 키워드 인자로 변경
+    5,  # 시드 값, 키워드 인자로 변경
+    1,  # 생성할 waveform의 수, 키워드 인자로 변경
     fn_index=1  # 함수 인덱스, 이미 키워드 인자로 되어 있음
     )
 
+    # 이미지 분류 결과를 포함한 전체 프롬프트 생성
+    full_prompt = f"{base_prompt} Inspired by: {image_classification_result}."
     # API 호출 결과 처리
     return result