Update app.py
Browse files
app.py
CHANGED
@@ -26,13 +26,14 @@ def generate_response(input_text, tokenizer, model):
|
|
26 |
input_ids = tokenizer.encode(input_text, return_tensors="pt", max_length=tokenizer.model_max_length, truncation=True)
|
27 |
|
28 |
# Modell használata a válasz generálásához
|
29 |
-
output = model.generate(input_ids,
|
30 |
|
31 |
# Decoding és válasz visszaadása
|
32 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
33 |
return response
|
34 |
|
35 |
|
|
|
36 |
# Dokumentum feltöltése drag and drop segítségével
|
37 |
document_file = st.file_uploader("Húzd ide a dokumentumot vagy kattints a feltöltéshez", type=["pdf", "docx", "doc"])
|
38 |
|
|
|
26 |
input_ids = tokenizer.encode(input_text, return_tensors="pt", max_length=tokenizer.model_max_length, truncation=True)
|
27 |
|
28 |
# Modell használata a válasz generálásához
|
29 |
+
output = model.generate(input_ids, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
|
30 |
|
31 |
# Decoding és válasz visszaadása
|
32 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
33 |
return response
|
34 |
|
35 |
|
36 |
+
|
37 |
# Dokumentum feltöltése drag and drop segítségével
|
38 |
document_file = st.file_uploader("Húzd ide a dokumentumot vagy kattints a feltöltéshez", type=["pdf", "docx", "doc"])
|
39 |
|