Create prompt_extend.py

prompt_extend.py

@@ -0,0 +1,7 @@
+from transformers import pipeline
+
+pipe = pipeline('text-generation', model='spaces/phenomenon1981/MagicPrompt-Stable-Diffusion')
+
+def extend_prompt(prompt):
+    return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]
+