Fred808 commited on
Commit
3542025
·
verified ·
1 Parent(s): 4c64189

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id)
20
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
21
 
22
  # Define the pre-prompt
23
- PRE_PROMPT = "You are a helpful virtual assistant. Answer the user's question clearly and concisely."
24
 
25
  # Define request body schema
26
  class TextGenerationRequest(BaseModel):
 
20
  pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
21
 
22
  # Define the pre-prompt
23
+ PRE_PROMPT = "You are a helpful customer service assistant. Answer the customer's question clearl, with care and concisely."
24
 
25
  # Define request body schema
26
  class TextGenerationRequest(BaseModel):