diginoron commited on
Commit
353aeae
·
verified ·
1 Parent(s): 20ee226

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- 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)
 
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)