Update app.py

app.py

@@ -131,6 +131,12 @@ def post_process_response(response):
     
     return response.strip()
 
+interaction_count = 0
+
+def cleanup_memory():
+    gc.collect()
+    torch.cuda.empty_cache()
+
 @spaces.GPU(duration=120)
 def generate(
     message: str,