Update app.py

app.py

@@ -840,6 +840,12 @@ def generate_script(topic):
     except Exception as e:
         print(f"Error during script generation: {e}")
         return "스크립트 생성 중 오류가 발생했습니다."
+
+
+def cleanup():
+    """메모리 정리 함수"""
+    torch.cuda.empty_cache()
+    gc.collect()
         
 # Gradio Interface Definition
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as iface: