Catch API error

app.py

@@ -272,12 +272,12 @@ def run_model(mode_set, user_input):
             inputs = inquire_prompt
             attempts = 0
             out = None
-            while out==None or attempts<MAX_API_CALLS: 
+            while out==None and attempts<MAX_API_CALLS: 
+                attempts += 1
                 try:
                     # api call 
                     out = inference(inputs,INFERENCE_PARAMS) 
                 except Exception:
-                    attempts += 1
                     if attempts<MAX_API_CALLS:
                         st.warning(f"HF Inference API call (attempt {attempts} of {MAX_API_CALLS}) has failed. Trying again...")
                     else: