m-ric HF staff commited on
Commit
1b6be6f
1 Parent(s): e8e18da

Update text_to_image.py

Browse files
Files changed (1) hide show
  1. text_to_image.py +1 -1
text_to_image.py CHANGED
@@ -41,7 +41,7 @@ class TextToImageTool(Tool):
41
 
42
  self.is_initialized = True
43
 
44
- def __call__(self, prompt):
45
  if not self.is_initialized:
46
  self.setup()
47
 
 
41
 
42
  self.is_initialized = True
43
 
44
+ def forward(self, prompt: str):
45
  if not self.is_initialized:
46
  self.setup()
47