Spaces:

Paridhim
/

llm_server

Sleeping

Paridhim commited on Apr 30, 2024

Commit

8bd982c

1 Parent(s): 85f13a7

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -10,7 +10,7 @@ llm_wrapper = LLMWrapper()
-@app.post("/")
 async def generate_text(request: Request):
     raw_data = await request.body()  # Get the raw body data from the request
     prompt = raw_data.decode('utf-8')

+@app.post("/llm_on_cpu")
 async def generate_text(request: Request):
     raw_data = await request.body()  # Get the raw body data from the request
     prompt = raw_data.decode('utf-8')