Update app.py

app.py

@@ -39,7 +39,7 @@ def check_key_gemini_availability(key, ai_model):
     except Exception as e:
         return False, f"Error: {e}"
 
-def check_keys(keys_text):
+def check_keys(keys_text, ai_model):
     """
     Checks a list of Gemini API keys and categorizes them into valid and invalid.
 
@@ -56,7 +56,7 @@ def check_keys(keys_text):
     invalid_keys = []
 
     for key in key_list:
-        is_valid, error_msg = check_key_gemini_availability(key)
+        is_valid, error_msg = check_key_gemini_availability(key, ai_model)
         if is_valid:
             valid_keys.append(key)
         else: