Update api_usage.py

api_usage.py

@@ -98,9 +98,9 @@ def format_status(list_models_avai, headers):
         if "error" in result:
             e = result.get("error", {}).get("code", "")
             if e == None:
-                rpm_num = int(response.headers.get("x-ratelimit-limit-requests", 0))
-                tpm_num = int(response.headers.get('x-ratelimit-limit-tokens', 0))
-                tpm_left = int(response.headers.get('x-ratelimit-remaining-tokens', 0))
+                rpm_num = int(r.headers.get("x-ratelimit-limit-requests", 0))
+                tpm_num = int(r.headers.get('x-ratelimit-limit-tokens', 0))
+                tpm_left = int(r.headers.get('x-ratelimit-remaining-tokens', 0))
                 _rpm = '{:,}'.format(rpm_num).replace(',', ' ')
                 _tpm = '{:,}'.format(tpm_num).replace(',', ' ')
                 _tpm_left = '{:,}'.format(tpm_left).replace(',', ' ')