Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ model.eval()
|
|
| 53 |
# Generare directΔ (fΔrΔ pipeline)
|
| 54 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 55 |
def chat_fn(message, history):
|
| 56 |
-
prompt = f"<|user|>\n{message.strip()}\n<|assistant|>\nAnswer the question clearly and concisely
|
| 57 |
inputs = tok(prompt, return_tensors="pt").to(model.device)
|
| 58 |
with torch.no_grad():
|
| 59 |
output = model.generate(
|
|
|
|
| 53 |
# Generare directΔ (fΔrΔ pipeline)
|
| 54 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 55 |
def chat_fn(message, history):
|
| 56 |
+
prompt = f"<|user|>\n{message.strip()}\n<|assistant|>\nAnswer the question clearly and concisely: SAY you dont know if you are not fine tuned with data related to that question\n"
|
| 57 |
inputs = tok(prompt, return_tensors="pt").to(model.device)
|
| 58 |
with torch.no_grad():
|
| 59 |
output = model.generate(
|