Update app.py

app.py

@@ -61,7 +61,7 @@ def generate(
     if torch.cuda.is_available():
 
         if use_img2img:
-            pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)
+            pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
             
             response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
@@ -106,6 +106,7 @@ def generate(
         images = pipe(
             prompt=prompt,
             image=init_image,
+            strength=0.5,
             negative_prompt=negative_prompt,
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,
@@ -114,7 +115,6 @@ def generate(
             guidance_scale=guidance_scale_base,
             num_inference_steps=num_inference_steps_base,
             generator=generator,
-            strength=0.5,
             output_type="pil",
         ).images[0]
         return images