Update app.py

app.py

@@ -290,16 +290,16 @@ def main():
             else:
                 report = raw_result
                 # report = extract_assistant_response(raw_result)
-            
+            clear_gpu_memory()
             progress_bar.progress(100)
             status_text.success("**✅ Generation complete!**")
             html("<script>localStorage.setItem('freezeTimer', 'true');</script>", height=0)
             st.session_state.timer_frozen = True
-    
+            
             # formatted_report = report.replace('\n', '<br>')
             
             st.subheader("Generated Report:")
             st.markdown(f"<div style='font-size: normal; font-weight: normal;'>{report}</div>", unsafe_allow_html=True)
-
+            
 if __name__ == '__main__':
     main()