Update main.py

main.py

@@ -16,7 +16,7 @@ class RequestBody(BaseModel):
 @app.post("/generate/")
 async def generate_text(request: RequestBody):
     try:
-        prompt = request.prompt
+        prompt = sf.encoder(request.prompt)
         input_ids = test_tokenizer(prompt, return_tensors='pt', truncation=False).input_ids
         outputs = test_model.generate(
             input_ids=input_ids,