Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
 from transformers import pipeline
 
-generator = pipeline('text-generation', model='AlekseyKorshuk/gpt2-jokes')
+generator = pipeline('text-generation', model='AlekseyKorshuk/pythia-1.4b-deduped-jokes')
 
 def generate(text):
-    result = generator(text, max_length=256, num_return_sequences=1, do_sample=True)
+    result = generator(text, max_new_tokens=256, num_return_sequences=1, do_sample=True)
     return result[0]["generated_text"]
 
 examples = [
-    ["The Moon's orbit around Earth has"],
-    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
+    ["Two guys walk into a bar."],
+    ["Why was the musician arrested?"],
 ]
 
 demo = gr.Interface(