mariusjabami commited on
Commit
2db33b9
·
verified ·
1 Parent(s): 88bfb08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,7 +48,8 @@ def generate_response(message, max_tokens, temperature, top_p):
48
  global stop_signal
49
  stop_signal = False
50
 
51
- prompt = f"Entrada: {message}\nResposta:"
 
52
  inputs = tokenizer(prompt, return_tensors="pt").to(device)
53
 
54
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
48
  global stop_signal
49
  stop_signal = False
50
 
51
+ prompt = f"Contexto: {message}\nResposta:"
52
+
53
  inputs = tokenizer(prompt, return_tensors="pt").to(device)
54
 
55
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)