Update app.py

app.py

@@ -1,3 +1,22 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load("models/AlekseyKorshuk/gpt2-jokes").launch()
+generator = pipeline('text-generation', model='AlekseyKorshuk/gpt2-jokes')
+
+def generate(text):
+    result = generator(text, max_length=30, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+examples = [
+    ["The Moon's orbit around Earth has"],
+    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
+]
+
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    examples=examples
+)
+
+demo.launch()