Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ def generate_response_with_gpt2(answer, user_question):
|
|
46 |
input_ids = gpt2_tokenizer.encode(prompt, return_tensors="pt", truncation=True)
|
47 |
|
48 |
# تولید خروجی از مدل GPT2
|
49 |
-
|
50 |
|
51 |
# تبدیل توکنها به متن
|
52 |
generated_text = gpt2_tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
|
|
46 |
input_ids = gpt2_tokenizer.encode(prompt, return_tensors="pt", truncation=True)
|
47 |
|
48 |
# تولید خروجی از مدل GPT2
|
49 |
+
output_ids = gpt2_model.generate(input_ids, max_new_tokens=80, num_return_sequences=1, no_repeat_ngram_size=2, top_p=0.95, temperature=0.7)
|
50 |
|
51 |
# تبدیل توکنها به متن
|
52 |
generated_text = gpt2_tokenizer.decode(output_ids[0], skip_special_tokens=True)
|