Update app.py

app.py

@@ -43,8 +43,12 @@ def load_models():
             "text-generation",
             model=story_model,
             tokenizer=story_tokenizer,
-            max_new_tokens=300,
-            temperature=0.7
+            max_new_tokens=230,
+            temperature=0.9,
+            top_k=50,
+            top_p=0.9,
+            repetition_penalty=1.1,
+            eos_token_id=151645
         )
 
         return caption_pipe, story_pipe