Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
|
@@ -2,9 +2,8 @@ from typing import Union
|
|
| 2 |
|
| 3 |
from fastapi import FastAPI
|
| 4 |
from epub2txt import epub2txt
|
| 5 |
-
import config
|
| 6 |
|
| 7 |
-
model = llm.model(
|
| 8 |
prompt = os.getenv("prompt")
|
| 9 |
|
| 10 |
from fastllm_pytools import llm
|
|
|
|
| 2 |
|
| 3 |
from fastapi import FastAPI
|
| 4 |
from epub2txt import epub2txt
|
|
|
|
| 5 |
|
| 6 |
+
model = llm.model(os.getenv("checkpoint_path"))
|
| 7 |
prompt = os.getenv("prompt")
|
| 8 |
|
| 9 |
from fastllm_pytools import llm
|