phuongnv commited on
Commit
4c4aeca
1 Parent(s): eeddbd0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -13,7 +13,7 @@ test_model = Llama(model_name)
13
  class RequestBody(BaseModel):
14
  prompt: str
15
 
16
- @app.get("/generate/")
17
  async def generate_text(request: RequestBody):
18
  try:
19
  prompt = sf.encoder(request.prompt)
 
13
  class RequestBody(BaseModel):
14
  prompt: str
15
 
16
+ @app.post("/generate/")
17
  async def generate_text(request: RequestBody):
18
  try:
19
  prompt = sf.encoder(request.prompt)