Update app.py

app.py

@@ -71,11 +71,13 @@ def king(type ,
         progress=gr.Progress(track_tqdm=True),
     ):
     if type=="Image Editing" :
-        raw_image = Image.open(input_image)
+        raw_image = Image.open(input_image).convert('RGB')
         inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)
-        out = model.generate(**inputs, min_length=10, max_length=25)
+        pixel_values = inputs.pixel_values
+        out = model.generate(pixel_values=pixel_values, max_length=25, num_beams=3, repetition_penalty=2.5)
         caption = processor.decode(out[0], skip_special_tokens=True)
         instructions = f"{instruction} {caption} {instruction}"
+        print(instructions)
         if randomize_seed:
             seed = random.randint(0, 99999)
         text_cfg_scale = text_cfg_scale
@@ -193,7 +195,7 @@ with gr.Blocks(css=css) as demo:
             generate_button = gr.Button("Generate")
 
     with gr.Row():
-        input_image = gr.Image(label="Image", type="pil", interactive=True)
+        input_image = gr.Image(label="Image", type='filepath', interactive=True)
 
     with gr.Row():
         text_cfg_scale = gr.Number(value=7.3, step=0.1, label="Text CFG", interactive=True)