Toilatop1sever commited on
Commit
b73965e
·
verified ·
1 Parent(s): fca2cfa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=4096, n_threads=4)
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!"}