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