Update text_to_image.py

text_to_image.py

@@ -40,12 +40,12 @@ class TextToImageTool(Tool):
 
         self.is_initialized = True
 
-    def __call__(self, prompt):
+    def __call__(self, prompt, num_inference_steps=25):
         if not self.is_initialized:
             self.setup()
 
         negative_prompt = "low quality, bad quality, deformed, low resolution"
         added_prompt = " , highest quality, highly realistic, very high resolution"
 
-        return self.pipeline(prompt + added_prompt, negative_prompt=negative_prompt, num_inference_steps=25).images[0]
+        return self.pipeline(prompt + added_prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps).images[0]