Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a text generation model (GPT-2 for simplicity)
-generator = pipeline('text-generation', model='gpt2')
+# Use DistilGPT-2 for faster generation
+generator = pipeline('text-generation', model='distilgpt2')
 
 def generate_character_description(name, gender, trait, superpower):
     prompt = f"Generate a detailed description of a character named {name}. They are {gender}, {trait}, and their superpower is {superpower}. Provide a short story or background for this character."
     # Generate the description based on the prompt
-    description = generator(prompt, max_length=200)[0]['generated_text']
+    description = generator(prompt, max_length=100)[0]['generated_text']
     return description
 
 # Gradio interface
@@ -15,7 +15,7 @@ iface = gr.Interface(
     fn=generate_character_description,
     inputs=[
         gr.Textbox(label="Name", placeholder="Enter the character's name"),
-        gr.Radio(label="Gender", choices=["male", "female", "non-binary"]),
+        gr.Radio(label="Gender", choices=["male", "female"]),
         gr.Textbox(label="Personality Trait", placeholder="e.g., brave, clever, kind"),
         gr.Textbox(label="Superpower", placeholder="e.g., flying, invisibility, super strength")
     ],