Update app.py

app.py

@@ -10,39 +10,25 @@ st.header("List Ollama model downloads")
 HOST = os.getenv("HOST", "https://1tsnakers-ollamasearchapi.hf.space")
 
 def is_api_up():
-    
-    r = requests.get(f"{HOST}/ping")
-    
-    return r.status_code == 200
+    try:
+        r = requests.get(f"{HOST}/ping")
+        return r.status_code == 200
+    except:
+        return False
 
 if st.button("Make list"):
     start_time = time.time()
+    
     if not is_api_up():
-        st.error(f"HTTP error {r.status_code}")
+        st.error("API is not reachable")
     else:
         r = requests.get(f"{HOST}/library")
+        library = r.json()["results"]
 
-        library = r.json()
-        library = library["results"]
-
-        # add model data
-        models = []
-        for model in library:
-            models.append(model["model_base_name"])
-
-        # add pull count
-        pull_count = []
-        for model in library:
-            pull_count.append(int(model["pull_count"]))
-
-        pull_count_str = []
-        for model in library:
-            pull_count_str.append(model["pull_count_str"])
-
-        # add date updated
-        date = []
-        for model in library:
-            date.append(datetime.fromisoformat(model["last_updated_iso"]))
+        models = [model["model_base_name"] for model in library]
+        pull_count = [int(model["pull_count"]) for model in library]
+        pull_count_str = [model["pull_count_str"] for model in library]
+        date = [datetime.fromisoformat(model["last_updated_iso"]) for model in library]
 
         df = pd.DataFrame({
             "model": models,
@@ -50,19 +36,16 @@ if st.button("Make list"):
             "pull_count_str": pull_count_str,
             "date": date
         })
-        end_time = time.time()
-        elapsed_time = end_time - start_time
-        
+
+        elapsed_time = time.time() - start_time
         st.write(f"{len(models)} models counted in {elapsed_time:.4f} seconds")
 
         st.dataframe(
-            dfstyle.format({"pull_count": lambda x, i=None: pull_count_str[i]}),
+            df[["model", "pull_count_str", "date"]],
             hide_index=True,
             column_config={
-                "name": "AI Model",
+                "model": "AI Model",
                 "pull_count_str": "Pulls",
-                "date": st.column_config.DateColumn(
-                    "Last Updated"
-                )
+                "date": st.column_config.DateColumn("Last Updated")
             }
         )