Update app.py

app.py

@@ -7,8 +7,9 @@ from threading import Timer
 HUGGINGFACE_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
 def get_available_free():
     models_dict = InferenceClient(token=HUGGINGFACE_TOKEN).list_deployed_models("text-generation-inference")
-    models = models_dict['text-generation'] + models_dict['text2text-generation'] #token=HUGGINGFACE_TOKEN
+    models = models_dict['text-generation'] + models_dict['text2text-generation']
     models_vision = models_dict['image-text-to-text']
+    models_others = InferenceClient(token=HUGGINGFACE_TOKEN).list_deployed_models("text-generation")
     
     models_conclusion = {
         "Model": [],
@@ -17,7 +18,7 @@ def get_available_free():
         "Chat Completion": [],
         "Vision": []
     }
-    for m in models + models_vision:
+    for m in list(set(models + models_vision + models_others)):
         text_available = False
         chat_available = False
         vision_available = False
@@ -25,7 +26,7 @@ def get_available_free():
             vision_available = True
         pro_sub = False
         try:
-            InferenceClient(m, timeout=10, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
+            InferenceClient(m, timeout=60, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
             text_available = True
         except Exception as e:
             print(e)
@@ -38,7 +39,7 @@ def get_available_free():
                     return pd.read_csv("data.csv").to_dict(orient='list')
                 return []
         try:
-            InferenceClient(m, timeout=10).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1) #token=HUGGINGFACE_TOKEN
+            InferenceClient(m, timeout=60).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
             chat_available = True
         except Exception as e:
             print(e)