story to ui

app.py

@@ -60,7 +60,7 @@ def create_story(text_seed):
         MinLengthLogitsProcessor(10, eos_token_id=model.config.eos_token_id),
     ]
   )
-  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=20)])
+  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=80)])
 
   outputs = model.greedy_search(
     input_ids, logits_processor=logits_processor, stopping_criteria=stopping_criteria
@@ -129,7 +129,7 @@ image = gr.inputs.Image(type="pil")
 label = gr.outputs.Label(num_top_classes=5)
 examples = [["cats.jpg"]]
 title = "Generate a Story from an Image"
-description = "Demo for classifying images with Perceiver IO. To use it, simply upload an image and click 'submit', a caption is autogenerated as well"
+description = "Demo for classifying images with Perceiver IO. To use it, simply upload an image and click 'submit', a story is autogenerated as well"
 article = "<p style='text-align: center'></p>"
 
 img_info1 = gr.Interface(
@@ -143,7 +143,7 @@ img_info2 = gr.Interface(
     inputs=image,
     #outputs=label,
     outputs = [
-    gr.outputs.Textbox(label = 'Caption')
+    gr.outputs.Textbox(label = 'Story')
 ],
 )