Spaces:
Running
Running
File size: 776 Bytes
106db30 575e725 106db30 55fe9af 106db30 575e725 106db30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from llama_cpp.server.app import create_app, Settings
from fastapi.responses import HTMLResponse
from fastapi.middleware.gzip import GZipMiddleware
import os
app = create_app(
Settings(
n_threads=2, # set to number of cpu cores
model="model/gguf-model.bin",
embedding=True,
chat_format="zephyr"
)
)
app.add_middleware(GZipMiddleware, minimum_size=1000)
# Read the content of index.html once and store it in memory
with open("index.html", "r") as f:
content = f.read()
@app.get("/", response_class=HTMLResponse)
async def read_items():
return content
if __name__ == "__main__":
import uvicorn
uvicorn.run(app,
host=os.environ["HOST"],
port=int(os.environ["PORT"])
)
|