Update app.py

app.py

@@ -5,6 +5,13 @@ import logging
 import requests
 
 webhook_server = os.getenv('webhook_server')
+ZeroGPU = os.getenv('ZeroGPU')
+BASE_URL = os.getenv('SpaceURL')
+
+# Define your headers including the Authorization with your token
+headers = {
+    "Authorization": f"Bearer {API_TOKEN}"
+}
 
 # Initialize the client for image generation "prithivMLmods/IMAGINEO-4K" "mukaist/DALLE-4K"
 client_image = Client("measmonysuon/DALLE-4K")
@@ -27,7 +34,10 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     full_prompt = f"{prompt}, Canon EOS R5, 4K, Photo-Realistic, appearing photorealistic with super fine details, high resolution, natural look, hyper realistic photography, cinematic lighting, --ar 64:37, --v 6.0, --style raw, --stylize 750"
 
     try:
-        result = client_image.predict(
+        response = requests.post(
+            f"{BASE_URL}/run",
+            headers=headers,
+            json={
             prompt=full_prompt,
             negative_prompt="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
             use_negative_prompt=True,
@@ -36,13 +46,15 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
             width=width,
             height=height,
             guidance_scale=5,
-            randomize_seed=True,
-            api_name="/run"
+            randomize_seed=True
+        }
         )
         logger.info("Image generation successful.")
+        response.raise_for_status()
+        result = response.json()
         return result
-    except Exception as e:
-        logger.error(f"Error generating image: {e}")
+    except requests.RequestException as e:
+        print(f"Request exception: {e}")
         return None
 
 def request_otp(user_chat_id):