Upload app.py

app.py

@@ -4,18 +4,29 @@ import base64
 from PIL import Image
 from io import BytesIO
 
-API_URL = "https://huggingface.co/spaces/florafeng/UCLIP-IMG-GEN/"
+API_URL = "https://florafeng-uclip-img-gen.hf.space/run/predict"
 
 
 def generate(prompt, seed):
     response = requests.post(
         API_URL,
-        json={"prompt": prompt, "seed": int(seed)}
+        json={
+            "data": [prompt, int(seed)]
+        }
     )
 
-    data = response.json()
+    if response.status_code != 200:
+        return f"Error: {response.status_code} - {response.text}"
 
-    img_bytes = base64.b64decode(data["image"])
+    try:
+        data = response.json()
+    except Exception:
+        return f"Invalid JSON response: {response.text}"
+
+    # Gradio responses are usually in: data["data"]
+    img_base64 = data["data"][0]
+
+    img_bytes = base64.b64decode(img_base64)
     image = Image.open(BytesIO(img_bytes))
 
     return image