Create main.py

main.py

@@ -0,0 +1,21 @@
+import json
+import httpx
+from flask import Flask, jsonify
+
+app = Flask(__name__)
+
+@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)