Update app.py

app.py

@@ -38,7 +38,7 @@ model_id = "alaa-lab/InstructCV"
 
 def main():
     # pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cpu")
-    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, safety_checker=None).to("cuda")
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cuda")
     example_image = Image.open("imgs/example2.jpg").convert("RGB")
     
 
@@ -49,8 +49,8 @@ def main():
             example_instruction,
             seed,
             0,
-            text_cfg_scale,
-            image_cfg_scale,
+            text_cfg_scale: float,
+            image_cfg_scale: float,
         )
 
     def generate(