Update app.py

app.py

@@ -1,23 +1,25 @@
 import gradio as gr
 from transformers import pipeline
 
-used_model = 'openai-community/gpt2'
-
-pipe = pipeline(task='text-generation', model=used_model)
+pipeline = pipeline(task='text-generation', model='openai-community/gpt2')
 
 def generate_alternatives(text):
-  items = pipe(text, max_length=50, num_return_sequences=5)
+  items = pipeline(text, max_length=150, num_return_sequences=2)
   texts = []
   for item in items:
     texts.append(str(items.index(item) + 1) + '. ' + item['generated_text'])
-    texts.append(str('') + 150 * '-')
+    # separator
+    texts.append(str('') + 15 * '-')
   return '\n'.join(texts)
 
-gradio_app = gr.Interface(
-  generate_alternatives,
-  inputs="text", #Please insert the text that you want to enhance",
-  outputs="text" #The generated alternatives"
+# Gradio UI
+examples=[['Today I got out of bed and went to school']]
+ui = gr.Interface(
+    generate_alternatives,
+    inputs="text",
+    outputs="text",
+    title="GPT-2 Test - Zircon.tech",
+    description="Enter some text to let GPT-2 give enhanced version of it!",
+    examples=examples
 )
-
-if __name__ == "__main__":
-    gradio_app.launch()
+ui.launch(debug=True)