Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 pipe = pipeline('text-generation', model='daspartho/prompt-extend')
 
 def extend_prompt(input):
-    return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]
+    return pipe(input+',', num_return_sequences=1)[0]["generated_text"]
 
 input_prompt = gr.Text(label="Enter the initial prompt")
 sd2_output = gr.Text(label="Extended prompt suitable for Stable Diffusion 2")