Spaces:
Sleeping
Sleeping
Commit
·
684e066
1
Parent(s):
11ad5ec
Edited System Prompt
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def generate_response(user_input):
|
|
16 |
# Combiner le prompt système avec la question utilisateur
|
17 |
prompt = f"{system_prompt}\n\nUser: {user_input}\nNazareAI:"
|
18 |
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
|
19 |
-
outputs = model.generate(inputs["input_ids"], max_length=
|
20 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
21 |
return response
|
22 |
|
|
|
16 |
# Combiner le prompt système avec la question utilisateur
|
17 |
prompt = f"{system_prompt}\n\nUser: {user_input}\nNazareAI:"
|
18 |
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
|
19 |
+
outputs = model.generate(inputs["input_ids"], max_length=1024, num_return_sequences=1)
|
20 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
21 |
return response
|
22 |
|