Upload app.py

app.py

@@ -1,15 +1,14 @@
 import gradio as gr
 import requests
-import base64
 from PIL import Image
 from io import BytesIO
 
-API_URL = "https://florafeng-uclip-img-gen.hf.space/run/predict"
+API_URL = "https://YOUR-BACKEND-SPACE.hf.space/run/predict"
 
 
 def generate(prompt, seed):
     response = requests.post(
-        "https://florafeng-UCLIP-IMG-GEN.hf.space/run/predict",
+        API_URL,
         json={
             "data": [prompt, int(seed)]
         }
@@ -20,11 +19,15 @@ def generate(prompt, seed):
 
     data = response.json()
 
-    # Gradio typically returns base64 or a file path
     result = data["data"][0]
 
-    img_bytes = base64.b64decode(result)
-    image = Image.open(BytesIO(img_bytes))
+    # If backend returns a PIL image serialized as URL/path
+    if isinstance(result, str) and result.startswith("http"):
+        img_bytes = requests.get(result).content
+        image = Image.open(BytesIO(img_bytes))
+    else:
+        # fallback (Gradio usually handles this automatically)
+        return result
 
     return image
 
@@ -40,7 +43,6 @@ with gr.Blocks() as demo:
     seed = gr.Number(value=123, label="Seed")
 
     btn = gr.Button("Generate")
-
     output = gr.Image()
 
     btn.click(generate, inputs=[prompt, seed], outputs=output)