Update app.py

app.py

@@ -74,7 +74,7 @@ def king(type ,
         raw_image = Image.open(input_image).convert('RGB')
         inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)
         pixel_values = inputs.pixel_values
-        out = model.generate(pixel_values=pixel_values, max_length=25, num_beams=3, repetition_penalty=2.5)
+        out = model.generate(pixel_values=pixel_values, max_length=20, num_beams=3, repetition_penalty=2.5)
         caption = processor.decode(out[0], skip_special_tokens=True)
         instructions = f"{instruction} {caption} {instruction}"
         print(instructions)
@@ -86,7 +86,7 @@ def king(type ,
         steps=steps
         generator = torch.manual_seed(seed)
         output_image = pipe_edit(
-            instructions, image=input_image,
+            instructions, image=raw_image,
             guidance_scale=text_cfg_scale, image_guidance_scale=image_cfg_scale,
             num_inference_steps=steps, generator=generator, output_type="latent",
         ).images