Changes to be committed:

app.py

@@ -55,13 +55,9 @@ def image_to_image(prompt, image, dimension, flux_client):
     input_image=handle_file(padded_image_path),
     prompt=prompt,
     guidance_scale=2.5,
-    negative_prompt=None,
-    nag_negative_prompt="Low resolution, blurry, lack of details",
-    nag_scale=3,
     num_inference_steps=25,
     randomize_seed=True,
-    compare=False,
-    api_name="/sample"
+    api_name="/infer"
     )
 
     print(image)
@@ -70,7 +66,7 @@ def image_to_image(prompt, image, dimension, flux_client):
         os.remove(padded_image_path)
     except:
         pass
-    return image[0][1]
+    return image[0]
 
 def image_to_video(prompt, image, ltx_client):
     result = ltx_client.predict(
@@ -86,7 +82,7 @@ def image_to_video(prompt, image, ltx_client):
 
 def personalized_video(prompt, image, dimension, request: gr.Request):
     x_ip_token = request.headers['x-ip-token']
-    flux_client = Client("ChenDY/NAG_FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
+    flux_client = Client("black-forest-labs/FLUX.1-Kontext-Dev", headers={"x-ip-token": x_ip_token})
     ltx_client = Client("KingNish/ltx-video-distilled", headers={"x-ip-token": x_ip_token})
     image = image_to_image(prompt, image, dimension, flux_client)
     yield image, None