Update app.py

app.py

@@ -1,25 +1,19 @@
 import gradio as gr
 from gradio_client import Client, handle_file
-import requests
 from PIL import Image
-from io import BytesIO
 
-# Connect to external Hugging Face Space API
 hf_client = Client("multimodalart/cosxl")
 
-# Define your Gradio function
 def run_edit(image, prompt):
-    # Forward other fixed values to the API
     negative_prompt = "blurry, distorted"
     guidance_scale = 7
     steps = 20
 
-    # Save the uploaded PIL image
     image_path = "/tmp/uploaded_image.png"
     image.save(image_path)
 
-    # Send request to external model
-    result = hf_client.predict(
+    # Predict (returns file path)
+    result_path = hf_client.predict(
         image=handle_file(image_path),
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -28,29 +22,20 @@ def run_edit(image, prompt):
         api_name="/run_edit"
     )
 
-    # Extract result image URL and load
-    result_url = result.get("url")
-    if result_url is None:
-        raise ValueError("Failed to get image result from API.")
-
-    response = requests.get(result_url)
-    edited_image = Image.open(BytesIO(response.content)).convert("RGB")
+    # Load and return result
+    edited_image = Image.open(result_path).convert("RGB")
     return edited_image
 
-# Gradio interface: only ask for image and prompt
 interface = gr.Interface(
     fn=run_edit,
     inputs=[
         gr.Image(label="Upload Image", type="pil"),
-        gr.Textbox(label="Describe the edit (Prompt)", placeholder="e.g. Make the bus pink with clouds")
+        gr.Textbox(label="Prompt", placeholder="Describe what to edit")
     ],
     outputs=gr.Image(label="Edited Image"),
-    title="Smart Image Editor",
-    description="Upload an image and describe what to change. The model will edit it intelligently."
+    title="Image Editor using multimodalart/cosxl",
+    description="Uses Hugging Face model for real image editing based on prompt."
 )
 
-# Enable API call from Hugging Face client
 interface.api_name = "/run_edit"
-
-# Launch app
 interface.launch()