Update app.py

app.py

@@ -105,7 +105,7 @@ def process_response(response, soc_res, url, token):
             return []
 
     try:
-        res = response.json().get('img_array', []) if status_code == 202 else []
+        res = response.json().get('img_array', []) if status_code in [201, 202] else []
         if soc_res:
             res = soc_res[0] + res + soc_res[1]