eduard76 commited on
Commit
db8ea25
Β·
verified Β·
1 Parent(s): b61a9c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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:\n"
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(