Update app.py

app.py

@@ -31,7 +31,7 @@ def generate_image(prompt, aspect_ratio, realism):
     global TOKEN
     global tokens_tried
     global no_of_accounts
-
+    global model_id
     payload = {
         "id": model_id,  # Replace with appropriate ID if necessary
         "inputs": [prompt, aspect_ratio, str(realism).lower()],
@@ -43,7 +43,7 @@ def generate_image(prompt, aspect_ratio, realism):
         if "error" in response_data:
             if 'error 429' in response_data['error']:
                 if tokens_tried < no_of_accounts:
-                    token_id = (token_id + 1) % no_of_accounts
+                    token_id = (token_id + 1) % (no_of_accounts+1)
                     tokens_tried += 1
                     TOKEN = os.getenv(f"TOKEN{token_id}")
                     response_data = generate_image(prompt, aspect_ratio, realism)