Update app.py

app.py

@@ -1,9 +1,10 @@
 from flask import Flask, request, jsonify, send_from_directory
-from gradio_client import Client
-import base64, os, uuid
+import requests, base64, os
 
 app = Flask(__name__, static_folder="static")
-client = Client("mrfakename/Z-Image-Turbo")
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 
 @app.route("/")
 def index():
@@ -13,17 +14,16 @@ def index():
 def generate():
     data = request.json
     prompt = data.get("prompt", "a beautiful landscape")
-    try:
-        result = client.predict(
-            prompt=prompt,
-            api_name="/generate"
-        )
-        # result is a file path — read and base64 encode it
-        with open(result, "rb") as f:
-            img_b64 = base64.b64encode(f.read()).decode()
+    response = requests.post(
+        API_URL,
+        headers={"Authorization": f"Bearer {HF_TOKEN}"},
+        json={"inputs": prompt}
+    )
+    if response.status_code == 200:
+        img_b64 = base64.b64encode(response.content).decode()
         return jsonify({"image": img_b64})
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+    else:
+        return jsonify({"error": response.text}), 500
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)