Update app.py

app.py

@@ -103,10 +103,7 @@ os.environ['PYTORCH_NO_CUDA_MEMORY_CACHING'] = '1'
 # CUDA 초기화 방지
 torch.set_grad_enabled(False)
 
-def periodic_cleanup():
-    """주기적으로 실행될 메모리 정리 함수"""
-    clear_gpu_memory()
-    return None
+
 
 # Gradio 인터페이스에 주기적 정리 추가
 demo.load(periodic_cleanup, every=5)  # 5초마다 정리
@@ -418,6 +415,10 @@ footer {
 }
 """
 
+def periodic_cleanup():
+    """주기적으로 실행될 메모리 정리 함수"""
+    clear_gpu_memory()
+    return None
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.Markdown("""## Craft3D""")
@@ -611,6 +612,9 @@ 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()  # 큐 기능 활성화