Update random_charactor_generator.py

random_charactor_generator.py

@@ -8,7 +8,34 @@ class RandomCharatorGeneratorTool:
 
     outputs = ["json"]
 
+
+    def __init__(self, device=None, **hub_kwargs) -> None:
+        #if not is_accelerate_available():
+        #    raise ImportError("Accelerate should be installed in order to use tools.")
+
+        super().__init__()
+
+        self.device = device
+        self.pipeline = None
+        self.hub_kwargs = hub_kwargs
+
+        def setup(self):
+        #if self.device is None:
+        #    self.device = get_default_device()
+
+        #self.pipeline = DiffusionPipeline.from_pretrained(self.default_checkpoint)
+        #self.pipeline.scheduler = DPMSolverMultistepScheduler.from_config(self.pipeline.scheduler.config)
+        #self.pipeline.to(self.device)
+
+        #if self.device.type == "cuda":
+        #    self.pipeline.to(torch_dtype=torch.float16)
+
+        self.is_initialized = True
+
     def __call__(self, prompt):
+        if not self.is_initialized:
+            self.setup()
+        
         API_URL = "https://randomuser.me/api/"
         
         response = requests.get(API_URL)