Update app.py

app.py

@@ -1,4 +1,7 @@
-#TODO: verify seed works
+#TODO: remove options for custom input -> only default "trustworthy ai"
+#TODO: excatract from generated-text dict entry
+#TODO: add credits
+#TODO: add flagging option
 #TODO: integrate into markdown
 import gradio as gr
 from gradio import mix
@@ -31,7 +34,7 @@ def inference(text, seed):
     """
     #outtext = io2(text)
     set_seed(int(seed))
-    outtext = generator(text, max_length=100, num_return_sequences=1)
+    outtext = generator(text, max_length=100, num_return_sequences=1)['generated_text'] #get the string from the return dict with key 'generated text'
     return outtext