Update main.py
Browse files
main.py
CHANGED
@@ -13,7 +13,7 @@ test_model = Llama(model_name)
|
|
13 |
class RequestBody(BaseModel):
|
14 |
prompt: str
|
15 |
|
16 |
-
@app.
|
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)
|