CooLLaMACEO commited on
Commit
79fbde8
·
verified ·
1 Parent(s): 7d837d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,11 +30,11 @@ async def chat(request: Request, _ = Depends(verify_key)):
30
  body = await request.json()
31
  user_input = body.get("prompt", "")
32
 
33
- # This 'System Prompt' keeps the AI on track
34
- system_text = "Assistant is a helpful, polite AI. Assistant gives short and clear answers."
35
-
36
- # Formatting it for the MPT-7B model
37
- prompt = f"{system_text}\nUser: {user_input}\nAssistant:"
38
 
39
  response = llm(
40
  prompt,
 
30
  body = await request.json()
31
  user_input = body.get("prompt", "")
32
 
33
+ # Updated System Prompt in app.py
34
+ system_text = "Assistant is a polite and helpful AI named ChatMPT. It was created by the ChatMPT Team."
35
+
36
+ # This ensures that if you ask "Who made you?", it answers correctly.
37
+ prompt = f"### Instruction:\n{system_text}\n\n### Input:\n{user_input}\n\n### Response:\n"
38
 
39
  response = llm(
40
  prompt,