Update app.py

app.py

@@ -54,7 +54,13 @@ def generate_kids_story(character, setting, language):
 
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)
+    
+    # Delete model and associated objects 
     del generator
+    # Run garbage collection
+    gc.collect ()
+    # Empty CUDA cache
+    torch.cuda.empty_cache()
     
     return output[0]["generated_text"]