Update Garage/models/PowerPaintModel.py

Garage/models/PowerPaintModel.py

@@ -114,7 +114,7 @@ class PowerPaintModel:
         self.pipe.to(device)
         self.device = device
 
-    # @spaces.GPU(duration=120)
+    @spaces.GPU(duration=120)
     def __call__(
         self,
         input_image: Dict[str, Image.Image],
@@ -167,8 +167,8 @@ class PowerPaintModel:
         result = self.pipe(
             promptA=promptA,
             promptB=promptB,
-            promptU="cat",
-            # promptU=prompt,
+            # promptU="cat",
+            promptU=prompt,
             tradoff=fitting_degree,
             tradoff_nag=fitting_degree,
             image=Image.fromarray(np_inpimg.astype(np.uint8)).convert("RGB").convert("RGB"),