albhu commited on
Commit
252acf6
·
verified ·
1 Parent(s): b5a0f4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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, max_new_tokens=200, 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
  # 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