Test with getting first sentence

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 
-def greet(name):
-    generator = pipeline("text-generation")
-    return generator(name)
+def greet(input):
+    generator = pipeline("text-generation", model="distilgpt2")
+    generated_text = generator(input)[0]['generated_text']
+    # Extract the first sentence and format it
+    first_sentence = generated_text.split('.')[0].strip()
+    return f"'{first_sentence}'"
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()