Update app.py

app.py

@@ -130,7 +130,7 @@ def together_text_to_image(prompt: str):
             model="black-forest-labs/FLUX.1.1-pro", # Hardcoded model as requested
             width=1024,
             height=768,
-            steps=40, # Hardcoded steps as requested
+            steps=40,
             prompt=prompt,
         )
         return image_completion.data[0].url
@@ -156,19 +156,12 @@ def together_image_to_image(image_numpy, prompt: str):
         return "Please enter a prompt for image transformation."
 
     try:
-        # Convert the NumPy array image to a PIL Image
-        img = Image.fromarray(image_numpy.astype('uint8'), 'RGB')
-
-        # Convert the PIL Image to base64
-        buffered = io.BytesIO()
-        img.save(buffered, format="PNG")
-        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
         image_completion = client.images.generate(
             model="black-forest-labs/FLUX.1-kontext-max", # Hardcoded model as requested
             steps=40, # Hardcoded steps as requested
             prompt=prompt,
-            image_base64=img_base64
+            image_url=image_numpy
         )
         return image_completion.data[0].url
     except Exception as e:
@@ -248,7 +241,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
         gr.Markdown("Transform an uploaded image based on a text prompt using Together AI.")
         gr.Warning("This requires setting the TOGETHER_API_KEY environment variable.")
         with gr.Row():
-            together_image_input = gr.Image(label="Upload or paste an image", scale=2)
+            together_image_input = gr.Image(label="Upload or paste an image", type="filepath", scale=2)
             together_image_to_image_prompt = gr.Textbox(label="Enter your transformation prompt", scale=2)
             together_image_to_image_button = gr.Button("Transform Image", scale=1)
         together_image_to_image_output = gr.Image(label="Transformed Image (URL)", type="filepath", interactive=False)