Update app.py
Browse files
app.py
CHANGED
|
@@ -68,6 +68,10 @@ def build_prompt(user_text: str) -> str:
|
|
| 68 |
return f"<|system|>\n{SYSTEM_PROMPT}\n<|user|>\n{user_text}\n<|assistant|>\n"
|
| 69 |
|
| 70 |
# ---------- API endpoint ----------
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
@app.post("/generate")
|
| 72 |
def generate(req: PromptRequest):
|
| 73 |
llm_instance = get_llm() # lazy load
|
|
|
|
| 68 |
return f"<|system|>\n{SYSTEM_PROMPT}\n<|user|>\n{user_text}\n<|assistant|>\n"
|
| 69 |
|
| 70 |
# ---------- API endpoint ----------
|
| 71 |
+
@app.get("/")
|
| 72 |
+
def root():
|
| 73 |
+
return {"status": "ok"}
|
| 74 |
+
|
| 75 |
@app.post("/generate")
|
| 76 |
def generate(req: PromptRequest):
|
| 77 |
llm_instance = get_llm() # lazy load
|