Updated error handling

app.py

@@ -13,6 +13,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
+
 def get_mask(img_in):
 
     # print(f"{img_in=}")
@@ -28,18 +29,16 @@ def get_mask(img_in):
 
     response = requests.post(upload_url, headers=headers, files=files)
 
-    print(f"{response.status_code=}")
-
     if response.status_code == 200:
         result = response.json()
 
         mask = Image.open(io.BytesIO(base64.b64decode(
             result['mask']))).resize(img_in.size)
-        
+
         return (mask, result['emotion'])
     else:
-        print(f"ERROR: {response.text}")
-        raise gr.Error(response.text)
+        print(f"ERROR: {response.json()['detail']}")
+        raise gr.Error(response.json()['detail'])
 
 
 footer = r"""
@@ -67,4 +66,5 @@ with gr.Blocks(title="ARKA + TikTok Demo") as app:
     with gr.Row():
         gr.HTML(footer)
 
-app.launch(share=False, debug=True, enable_queue=True, show_error=True, auth=((os.environ.get("USERNAME"), os.environ.get("PASSWORD"))))
+app.launch(share=False, debug=True, enable_queue=True, show_error=True,
+           auth=((os.environ.get("USERNAME"), os.environ.get("PASSWORD"))))