ajaynagotha commited on
Commit
734cf6a
·
verified ·
1 Parent(s): 9f75948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def answer_question(question):
35
  context = " ".join([item.get('Text', '') for item in ds['train']])
36
  logger.info(f"Combined context length: {len(context)} characters")
37
  logger.info("Tokenizing input")
38
- inputs = tokenizer.encode_plus(question, context, return_tensors="pt", max_length=514, truncation=True)
39
  logger.info(f"Input tokens shape: {inputs['input_ids'].shape}")
40
  logger.info("Getting model output")
41
  outputs = model(**inputs)
 
35
  context = " ".join([item.get('Text', '') for item in ds['train']])
36
  logger.info(f"Combined context length: {len(context)} characters")
37
  logger.info("Tokenizing input")
38
+ inputs = tokenizer.encode_plus(question, context, return_tensors="pt", max_length=512, truncation=True)
39
  logger.info(f"Input tokens shape: {inputs['input_ids'].shape}")
40
  logger.info("Getting model output")
41
  outputs = model(**inputs)