alaselababatunde commited on
Commit
d0b98b9
·
1 Parent(s): 02f04dd
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -46,7 +46,7 @@ try:
46
  model=MODEL_NAME,
47
  device=DEVICE,
48
  max_new_tokens=1024,
49
- temperature=0.7,
50
  do_sample=True,
51
  top_p=0.9,
52
  repetition_penalty=1.2
@@ -112,7 +112,7 @@ class QueryInput(BaseModel):
112
  # ===============================================
113
  @app.get("/")
114
  async def root():
115
- return {"message": "✅ Tech Disciples AI (LangChain Memory) is running."}
116
 
117
  @app.post("/ai-chat")
118
  async def ai_chat(data: QueryInput, x_api_key: str = Header(None)):
 
46
  model=MODEL_NAME,
47
  device=DEVICE,
48
  max_new_tokens=1024,
49
+ temperature=0.5,
50
  do_sample=True,
51
  top_p=0.9,
52
  repetition_penalty=1.2
 
112
  # ===============================================
113
  @app.get("/")
114
  async def root():
115
+ return {"message": "✅ Tech Disciples AI is running."}
116
 
117
  @app.post("/ai-chat")
118
  async def ai_chat(data: QueryInput, x_api_key: str = Header(None)):