Update app.py

app.py

@@ -88,17 +88,13 @@ musicgen.set_generation_params(duration=10)
 # 6. 파이프라인 함수
 # ─────────────────────────────────────────────────────────────
 def generate_caption(image: Image.Image) -> str:
-    with torch.no_grad():                               # ★ 메모리 절약
-        pixel_values = feature_extractor(
-            images=image, return_tensors="pt"
-        ).pixel_values
-        output_ids = caption_model.generate(
-            pixel_values.to(caption_model.device),       # CPU 디바이스 통일
-            max_length=50
-        )
+    with torch.no_grad():
+        pixel_values = feature_extractor(images=image, return_tensors="pt").pixel_values
+        output_ids = caption_model.generate(pixel_values, max_length=50)
     return tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
 
+
 def generate_music(prompt: str) -> str:
     wav = musicgen.generate([prompt])           # batch size = 1
     tmpdir = tempfile.mkdtemp()