Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ if not os.path.exists(MODEL_FILE):
|
|
| 15 |
print("Download done!")
|
| 16 |
|
| 17 |
print("Loading model...")
|
| 18 |
-
llm = Llama(model_path=MODEL_FILE, n_ctx=
|
| 19 |
print("Model ready!")
|
| 20 |
|
| 21 |
class ChatRequest(BaseModel):
|
|
@@ -32,4 +32,4 @@ async def chat(req: ChatRequest):
|
|
| 32 |
|
| 33 |
@app.get("/")
|
| 34 |
async def root():
|
| 35 |
-
return {"status": "ok", "message": "AI Coder is running!"}
|
|
|
|
| 15 |
print("Download done!")
|
| 16 |
|
| 17 |
print("Loading model...")
|
| 18 |
+
llm = Llama(model_path=MODEL_FILE, n_ctx=2048, n_threads=4)
|
| 19 |
print("Model ready!")
|
| 20 |
|
| 21 |
class ChatRequest(BaseModel):
|
|
|
|
| 32 |
|
| 33 |
@app.get("/")
|
| 34 |
async def root():
|
| 35 |
+
return {"status": "ok", "message": "AI Coder is running!"}
|