Update main.py

main.py

@@ -1,21 +1,26 @@
-import json
-import httpx
-from flask import Flask, jsonify
+import aiohttp
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
 
-app = Flask(__name__)
+app = FastAPI()
 
-@app.route('/v1/models')
-def index():
-    modelsraw = httpx.get('https://huggingface.co/models-json?inference=warm&sort=trending&withCount=true')
-    models = json.loads(modelsraw.text)
-    output = []
-    for key in models['models']:
-        if key['private'] == False and key['gated'] == False:
-            output.append({
-                "created": 0,
-                "id": key['id'],
-                "object": "model",
-                "owned_by": key['author'],
-                "pipeline": key['pipeline_tag']
-            })
-    return jsonify(output)
+@app.get("/v1/models")
+async def index():
+    url = 'https://huggingface.co/models-json?inference=warm&sort=trending&withCount=true'
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url) as response:
+            models = await response.json()
+    
+    output = [
+        {
+            "created": 0,
+            "id": key['id'],
+            "object": "model",
+            "owned_by": key['author'],
+            "pipeline": key['pipeline_tag']
+        }
+        for key in models['models']
+        if not key['private'] and not key['gated']
+    ]
+    
+    return JSONResponse(content=output)