Update app.py

app.py

@@ -1,4 +1,5 @@
 #TODO: integrate into markdown
+#TODO: clean up output formatting
 import gradio as gr
 from gradio import mix
 from transformers import pipeline, set_seed
@@ -7,7 +8,7 @@ generator = pipeline('text-generation', model='gpt2')
 
 def inference(text, seed):
     set_seed(int(seed))
-    outtext = generator(text, max_length=100, num_return_sequences=1)['generated_text'] #get the string from the return dict with key 'generated text'
+    outtext = generator(text, max_length=100, num_return_sequences=1) #get the string from the return dict with key 'generated text'
     return outtext   
     
 gr.Interface(