Update main.py
Browse files
main.py
CHANGED
@@ -2,7 +2,6 @@ import os
|
|
2 |
import sys
|
3 |
import time
|
4 |
import signal
|
5 |
-
import psutil
|
6 |
import io
|
7 |
|
8 |
from fastapi import FastAPI, Request, status, Form, UploadFile
|
@@ -33,7 +32,7 @@ fn.load_model()
|
|
33 |
@app.post("/run")
|
34 |
async def api_run(text: str):
|
35 |
try:
|
36 |
-
|
37 |
|
38 |
return {"result": output}
|
39 |
except Exception as e:
|
|
|
2 |
import sys
|
3 |
import time
|
4 |
import signal
|
|
|
5 |
import io
|
6 |
|
7 |
from fastapi import FastAPI, Request, status, Form, UploadFile
|
|
|
32 |
@app.post("/run")
|
33 |
async def api_run(text: str):
|
34 |
try:
|
35 |
+
output = fn.fn_pke(text)
|
36 |
|
37 |
return {"result": output}
|
38 |
except Exception as e:
|