Update app.py
Browse files
app.py
CHANGED
|
@@ -67,7 +67,7 @@ def health():
|
|
| 67 |
return {"status": "ok"}
|
| 68 |
|
| 69 |
# Chat Completion API
|
| 70 |
-
@app.
|
| 71 |
async def complete(gen:GenModel):
|
| 72 |
try:
|
| 73 |
st = time()
|
|
@@ -101,7 +101,7 @@ async def complete(gen:GenModel):
|
|
| 101 |
)
|
| 102 |
|
| 103 |
# Chat Completion API
|
| 104 |
-
@app.
|
| 105 |
async def complete(
|
| 106 |
question: str,
|
| 107 |
system: str = "You are a story writing assistant.",
|
|
|
|
| 67 |
return {"status": "ok"}
|
| 68 |
|
| 69 |
# Chat Completion API
|
| 70 |
+
@app.post("/generate/")
|
| 71 |
async def complete(gen:GenModel):
|
| 72 |
try:
|
| 73 |
st = time()
|
|
|
|
| 101 |
)
|
| 102 |
|
| 103 |
# Chat Completion API
|
| 104 |
+
@app.get"/generate_stream")
|
| 105 |
async def complete(
|
| 106 |
question: str,
|
| 107 |
system: str = "You are a story writing assistant.",
|