Modify prompt

app.py

@@ -39,9 +39,10 @@ def fn(sign, cat):
     seed = random.randint(100, 1000000)
     set_seed(seed)
 
-    response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
-    pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
-    image = pipe(response[0]["generated_text"], num_inference_steps=5).images[0]
+    # response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
+    # pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
+    # image = pipe(response[0]["generated_text"], num_inference_steps=5).images[0]
+    image = pipe(sign + " art", num_inference_steps=5).images[0]
     return [image, starting_text]