Update app.py

app.py

@@ -33,7 +33,7 @@ def generate_story(caption):
     
     result = story_generator(
         prompt, 
-        max_length=100,
+        max_new_tokens=100,
         num_return_sequences=1,
         do_sample=True,
         temperature=1.0