Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def generate_response(message, max_tokens, temperature, top_p):
|
|
48 |
global stop_signal
|
49 |
stop_signal = False
|
50 |
|
51 |
-
prompt = f"{message}"
|
52 |
|
53 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
54 |
|
|
|
48 |
global stop_signal
|
49 |
stop_signal = False
|
50 |
|
51 |
+
prompt = f"Contexto: {message}\nResposta curta e direta:"
|
52 |
|
53 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
54 |
|