oleh13 commited on
Commit
a7c9ae3
·
verified ·
1 Parent(s): aed6d45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ class CompletionRequest(BaseModel):
18
  max_tokens: int = 128
19
  temperature: float = 0.0
20
 
21
- @app.post("/v1/completions")
22
  async def text_completions(request: CompletionRequest):
23
  try:
24
  inputs = tokenizer(request.prompt, return_tensors="pt").to(device)
 
18
  max_tokens: int = 128
19
  temperature: float = 0.0
20
 
21
+ @app.post("/chat/completions")
22
  async def text_completions(request: CompletionRequest):
23
  try:
24
  inputs = tokenizer(request.prompt, return_tensors="pt").to(device)