|
|
import aiohttp |
|
|
from fastapi import FastAPI |
|
|
from fastapi.responses import JSONResponse, PlainTextResponse |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
@app.get('/', response_class=PlainTextResponse) |
|
|
async def homepage(): |
|
|
return "Request or go to /v1/models to use the space, make sure its a GET" |
|
|
|
|
|
@app.get("/v1/models") |
|
|
async def main(): |
|
|
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) |
|
|
|
