Update app.py

app.py

@@ -5,24 +5,23 @@ import cv2
 import os
 
 def image_to_video(image):
-    image_array = np.array(image)
+    image_array = np.array(image.convert('RGB'))  # RGB로 변환
+
     output_path = '/mnt/data/output_video.mp4'
-    
-    height, width, layers = image_array.shape
+    height, width, _ = image_array.shape
     size = (width, height)
-    video = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), 30, size)
-    
+    fourcc = cv2.VideoWriter_fourcc(*'X264')  # X264 코덱 사용
+
+    video = cv2.VideoWriter(output_path, fourcc, 30, size)  # 프레임 속도를 30으로 설정
+
     for _ in range(150):  # 총 150 프레임 생성
         video.write(image_array)
-    
-    video.release()
-    
-    return output_path
 
+    video.release()
 
+    return output_path
 
 def setup_interface():
-    # Gradio 인터페이스 구성
     with gr.Blocks() as demo:
         gr.Markdown("### 이미지를 업로드하면 5초짜리 비디오를 생성합니다.")
         
@@ -34,6 +33,5 @@ def setup_interface():
     
     return demo
 
-# 인터페이스 실행
 demo = setup_interface()
 demo.launch()