pendar02 commited on
Commit
2c2de78
·
verified ·
1 Parent(s): d16f597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -127,7 +127,9 @@ def generate_summary(text, model, tokenizer):
127
 
128
  def generate_focused_summary(question, abstracts, model, tokenizer):
129
  """Generate focused summary based on question"""
130
- combined_input = f"Question: {question} Abstracts: " + " [SEP] ".join(abstracts)
 
 
131
 
132
  inputs = tokenizer(combined_input, return_tensors="pt", max_length=1024, truncation=True)
133
 
 
127
 
128
  def generate_focused_summary(question, abstracts, model, tokenizer):
129
  """Generate focused summary based on question"""
130
+ # Preprocess each abstract
131
+ formatted_abstracts = [preprocess_text(abstract) for abstract in abstracts]
132
+ combined_input = f"Question: {question} Abstracts: " + " [SEP] ".join(formatted_abstracts)
133
 
134
  inputs = tokenizer(combined_input, return_tensors="pt", max_length=1024, truncation=True)
135