Take attention on task state

server.py

@@ -79,9 +79,11 @@ def check_significance_wait_for_result(result_url):
             time.sleep(5)
         else:
             raise RuntimeError(f'Failed to get result. Status code: {response.status_code}')  # TODO: try-except do raise gr.error
-    
-    print(result)
-    return result['result']
+
+    if result["state"] == "COMPLETED":
+        return result['result']
+    else:
+        raise RuntimeError(result['result']['error'])
 
 def check_significance(model_a_path, model_b_path):
     result_url = check_significance_send_task(model_a_path, model_b_path)