Update app.py

app.py

@@ -4,8 +4,8 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='huggingtweets/wwm_shakespeare')
 
 def greet(name):
-    #return generator(name)[0]['generated_text']
-    return generator(name)
+    return generator(name)[0]['generated_text']
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(fn=greet, inputs="text", outputs="text", layout='vertical',
+title='Shakespeare Love Notes Generator', description='Input some words, and generate some love notes. Check the Linked Models for more detail.')
 iface.launch()