Added titlecase

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from titlecase import titlecase
 
 from transformers import BloomTokenizerFast, BloomForCausalLM
 tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-560m")
@@ -19,7 +20,7 @@ def generate(text, temp=0.7, logging=True):
     decoded = tokenizer.decode(output[0], skip_special_tokens=True)
     if logging:
         print(f"\n\n{'-'*100}\nInput:  {text}\nOutput: {decoded}\nTemp:   {temp}")
-    return decoded
+    return titlecase(decoded)
 
 description = "Generate Titles for the Vice Youtube Channel"
 title = "Vice Headlines"