Update app.py

app.py

@@ -298,6 +298,10 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
         clear_gpu_memory()
         return None, None
 
+        return generated_image
+    finally:
+        clear_gpu_memory()  # 함수 종료 시 메모리 정리
+
 def clear_gpu_memory():
     """GPU 메모리를 더 철저하게 정리하는 함수"""
     try:
@@ -409,6 +413,10 @@ def text_to_image(prompt: str, height: int, width: int, steps: int, scales: floa
             torch.cuda.synchronize()
         gc.collect()
 
+        return generated_image
+    finally:
+        clear_gpu_memory()  # 함수 종료 시 메모리 정리
+        
 css = """
 footer {
     visibility: hidden;
@@ -612,9 +620,6 @@ if __name__ == "__main__":
                 print("Warning: trellis_pipeline is None")
         except Exception as e:
             print(f"Warning: Initial preprocessing test failed: {e}")
-        
-        # 주기적 정리 설정
-        demo.load(periodic_cleanup, every=5)  # 5초마다 정리
             
         # Gradio 인터페이스 실행
         demo.queue()  # 큐 기능 활성화