Update app.py

app.py

@@ -7,7 +7,7 @@ from threading import Timer
 HUGGINGFACE_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
 def get_available_free(use_cache = False):
     if use_cache:
-        if os.path.exists("data.csv"):
+        if os.path.exists("./data.csv"):
             print("Loading data from file...")
             return pd.read_csv("data.csv").to_dict(orient='list')
     models_dict = InferenceClient(token=HUGGINGFACE_TOKEN).list_deployed_models("text-generation-inference")
@@ -30,7 +30,7 @@ def get_available_free(use_cache = False):
             vision_available = True
         pro_sub = False
         try:
-            InferenceClient(m, timeout=60, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
+            InferenceClient(m, timeout=120, token=HUGGINGFACE_TOKEN).text_generation("Hi.", max_new_tokens=1)
             text_available = True
         except Exception as e:
             print(e)
@@ -40,10 +40,10 @@ def get_available_free(use_cache = False):
                 print("Rate Limited!!")
                 if os.path.exists("data.csv"):
                     print("Loading data from file...")
-                    return pd.read_csv("data.csv").to_dict(orient='list')
+                    return pd.read_csv("./data.csv").to_dict(orient='list')
                 return []
         try:
-            InferenceClient(m, timeout=60).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
+            InferenceClient(m, timeout=120).chat_completion(messages=[{'role': 'user', 'content': 'Hi.'}], max_tokens=1)
             chat_available = True
         except Exception as e:
             print(e)
@@ -60,7 +60,7 @@ def get_available_free(use_cache = False):
         models_conclusion["Chat Completion"].append("---" if (pro_sub or (not chat_available and not text_available)) else ("✓" if chat_available else "⌀"))
         models_conclusion["Text Completion"].append("---" if (pro_sub or (not chat_available and not text_available)) else ("✓" if text_available else "⌀"))
         models_conclusion["Vision"].append("✓" if vision_available else "⌀")
-    pd.DataFrame(models_conclusion).to_csv("data.csv", index=False)
+    pd.DataFrame(models_conclusion).to_csv("./data.csv", index=False)
     return models_conclusion
 
 def update_data(use_cache = False):