Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 from utils.run_pipeline_and_save import run_pipeline_and_save
 from utils.zip_output import zip_output
 
+import os
+os.environ["PYTORCH_SDP_ATTENTION"] = "math"
+
 def run_agents(prompt):
     if not prompt.strip():
         return "Please enter a prompt.", None, None