Update app.py
Browse files
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
|
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):
|