Update app.py

app.py

@@ -73,8 +73,7 @@ def king(type ,
     if type=="Image Editing" :
         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=20, num_beams=3, repetition_penalty=2.5)
+        out = model.generate(**inputs, min_length=10, max_length=20)
         caption = processor.decode(out[0], skip_special_tokens=True)
         instructions = f"{instruction} {caption} {instruction}"
         print(instructions)
@@ -92,7 +91,7 @@ def king(type ,
         ).images         
 
         refine = refiner(
-            prompt=instruction2,
+            prompt=instructions,
             guidance_scale=guidance_scale,
             num_inference_steps=steps,
             image=output_image,