Update app.py

app.py

@@ -31,8 +31,10 @@ def loop_query_data():
         if m in models_vision:
             vision_available = True
         pro_sub = False
-        while True:
+        run_text = True
+        while run_text:
             try:
+                run_text = False
                 InferenceClient(m, timeout=10, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
                 text_available = True
             except Exception as e:
@@ -42,10 +44,11 @@ def loop_query_data():
                 if e and "Rate limit reached" in str(e):
                     print("Rate Limited, waiting 1 hour...")
                     time.sleep(60*60)
-                else:
-                    break
-        while True:
+                    run_text = True
+        run_chat = True
+        while run_chat:
             try:
+                run_chat = False
                 InferenceClient(m, timeout=10).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
                 chat_available = True
             except Exception as e:
@@ -55,8 +58,7 @@ def loop_query_data():
                 if e and "Rate limit reached" in str(e):
                     print("Rate Limited, waiting 1 hour...")
                     time.sleep(60*60)
-                else:
-                    break
+                    run_chat = True
         models_conclusion["Model"].append(m)
         models_conclusion["API"].append("Free" if chat_available or text_available else ("Pro Subscription" if pro_sub else "Not Responding"))
         models_conclusion["Chat Completion"].append("---" if (pro_sub or (not chat_available and not text_available)) else ("✓" if chat_available else "⌀"))