Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_client import Client
 from gradio_client import Client, handle_file
 import os
 import logging
@@ -58,23 +59,15 @@ def load_gallery():
         return [(item["video"], item["prompt"]) for item in reversed(gallery)]
     return []
 
+
 def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
     logging.info(f"Received prompt: {prompt}, steps: {steps}, cfg_scale: {cfg_scale}, "
                  f"eta: {eta}, fs: {fs}, seed: {seed}, video_length: {video_length}")
     
     try:
-        # 이미지 처리
-        if isinstance(image, str):  # 파일 경로인 경우
-            image_file = File(image)
-        else:  # numpy array인 경우
-            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
-                img = Image.fromarray(image.astype('uint8'), 'RGB')
-                img.save(temp_file.name)
-                image_file = File(temp_file.name)
-
         # 비디오 생성 요청
         result = api_client.predict(
-            image_file,
+            image,  # 직접 파일 경로 전달
             prompt,
             steps,
             cfg_scale,
@@ -86,10 +79,6 @@ def respond(image, prompt, steps, cfg_scale, eta, fs, seed, video_length):
         )
         logging.info("API response received: %s", result)
         
-        # 임시 파일 삭제 (numpy array였을 경우에만)
-        if not isinstance(image, str):
-            os.unlink(image_file.name)
-        
         # 결과 확인 및 처리
         if isinstance(result, str) and result.endswith('.mp4'):
             saved_video_path = save_to_gallery(result, prompt)