Update app.py

app.py

@@ -4,18 +4,16 @@ import os
 import logging
 import requests
 from huggingface_hub import login
+from huggingface_hub import whoami
 
 webhook_server = os.getenv('webhook_server')
 API_TOKEN = os.getenv('ZeroGPU')
 BASE_URL = os.getenv('SpaceURL')
+# Get user information
+user_info = whoami(token=API_TOKEN)
 
 # Set your token
-login(token='API_TOKEN', add_to_git_credential=True)
-
-# Define your headers including the Authorization with your token
-headers = {
-    "Authorization": f"Bearer {API_TOKEN}"
-}
+login(token=API_TOKEN, add_to_git_credential=True)
 
 # Initialize the client for image generation "prithivMLmods/IMAGINEO-4K" "mukaist/DALLE-4K"
 client_image = Client("measmonysuon/DALLE-4K")
@@ -38,28 +36,23 @@ 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:
-        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,
-            style:style,
-            seed:0,
-            width:width,
-            height:height,
-            guidance_scale:5,
-            randomize_seed:True
-        }
+        result = client_image.predict(
+            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,
+            style=style,
+            seed=0,
+            width=width,
+            height=height,
+            guidance_scale=5,
+            randomize_seed=True,
+            api_name="/run"
         )
         logger.info("Image generation successful.")
-        response.raise_for_status()
-        result = response.json()
         return result
-    except requests.RequestException as e:
-        print(f"Request exception: {e}")
-        return None
+    except Exception as e:
+        logger.error(f"Error generating image: {e}")
+        return Non
 
 def request_otp(user_chat_id):
     try: