output to string

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=80)])
+  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=200)])
 
   outputs = model.greedy_search(
     input_ids, logits_processor=logits_processor, stopping_criteria=stopping_criteria
@@ -106,6 +106,7 @@ def self_caption(image):
   #return(pred_dictionary)
   preds = ' '.join(preds)
   story = create_story(preds)
+  story = ' '.join(story)
   return story