Update app.py

app.py

@@ -31,26 +31,32 @@ def loop_query_data():
         if m in models_vision:
             vision_available = True
         pro_sub = False
-        try:
-            InferenceClient(m, timeout=10, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
-            text_available = True
-        except Exception as e:
-            # print(e)
-            if e and "Model requires a Pro subscription" in str(e):
-                pro_sub = True
-            if e and "Rate limit reached" in str(e):
-                print("Rate Limited, waiting 1 hour...")
-                time.sleep(60*60)
-        try:
-            InferenceClient(m, timeout=10).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
-            chat_available = True
-        except Exception as e:
-            # print(e)
-            if e and "Model requires a Pro subscription" in str(e):
-                pro_sub = True
-            if e and "Rate limit reached" in str(e):
-                print("Rate Limited, waiting 1 hour...")
-                time.sleep(60*60)
+        while True:
+            try:
+                InferenceClient(m, timeout=10, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
+                text_available = True
+            except Exception as e:
+                # print(e)
+                if e and "Model requires a Pro subscription" in str(e):
+                    pro_sub = True
+                if e and "Rate limit reached" in str(e):
+                    print("Rate Limited, waiting 1 hour...")
+                    time.sleep(60*60)
+                else:
+                    break
+        while True:
+            try:
+                InferenceClient(m, timeout=10).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
+                chat_available = True
+            except Exception as e:
+                # print(e)
+                if e and "Model requires a Pro subscription" in str(e):
+                    pro_sub = True
+                if e and "Rate limit reached" in str(e):
+                    print("Rate Limited, waiting 1 hour...")
+                    time.sleep(60*60)
+                else:
+                    break
         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 "⌀"))
@@ -175,7 +181,7 @@ with gr.Blocks() as demo:
     
     def update_every_two_hours(first_run):
         search_models(search_box.value, use_cache = first_run)
-        Timer(7200, update_every_two_hours, args=(False,)).start()
+        Timer(60, update_every_two_hours, args=(False,)).start()
     
     Timer(0, update_every_two_hours, args=(first_run,)).start()