Update app.py

app.py

@@ -10,13 +10,10 @@ def expand_prompt(prompt, num_variants=5, max_length=100):
     """
     Given a basic prompt, generate `num_variants` expanded prompts using GPT-J-6B.
     """
-    # Generate multiple completions using sampling.
     outputs = generator(prompt, max_length=max_length, num_return_sequences=num_variants, do_sample=True)
-    # Clean and join the outputs
     expanded = [out["generated_text"].strip() for out in outputs]
     return "\n\n".join(expanded)
 
-# Create a Gradio Interface
 iface = gr.Interface(
     fn=expand_prompt,
     inputs=gr.Textbox(lines=2, placeholder="Enter your basic prompt here...", label="Basic Prompt"),
@@ -31,4 +28,3 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
-