Update app.py

app.py

@@ -33,6 +33,9 @@ def load_model(model_name):
         top_k=50,        # Control diversity
         top_p=0.95       # Control diversity
     )
+    del model 
+    del tokenizer
+    
     return generator
 
 
@@ -51,7 +54,11 @@ def generate_kids_story(character, setting, language):
 
     messages = [{"role": "user", "content": prompt}]
     output = generator(messages)
+    del generator
+    
     return output[0]["generated_text"]
+
+
 css_style = """
     body {
         background-image: url('https://cdna.artstation.com/p/assets/images/images/074/776/904/large/pietro-chiovaro-r1-castle-chp.jpg?1712916847');