Update app.py

app.py

@@ -2,19 +2,18 @@
 #TODO: use transformer library directly
 import gradio as gr
 from gradio import mix
-#from transformers import pipeline, set_seed
+from transformers import pipeline, set_seed
 
 #title = "trustworthy artificial intelligence workshop - content generator"
 description = "based on the gpt2 demo interface by <a href='https://huggingface.co/spaces/docs-demos/gpt2/tree/main'>ahsen khaliq</a>"
 
 #io1 = gr.Interface.load("huggingface/distilgpt2")
 
-#generator = pipeline('text-generation', model='gpt2')
-#set_seed(42)
+generator = pipeline('text-generation', model='gpt2')
+set_seed(42)
 
 
 io2 = gr.Interface.load("huggingface/gpt2-large")
-#TODO: 1) seed 2) output lenght
 
 #io3 = gr.Interface.load("huggingface/gpt2-medium")
 
@@ -31,8 +30,8 @@ def inference(text):
     else:
         outtext = io1(text)
     """
-    outtext = io2(text, max_length=60)
-    #outtext = generator(text, max_length=30, num_return_sequences=5)
+    #outtext = io2(text)
+    outtext = generator(text, max_length=30, num_return_sequences=5)
     return outtext