Chri12345 commited on
Commit
d476eec
·
verified ·
1 Parent(s): e8c4dfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,13 +16,13 @@ def load_model():
16
 
17
  # Cache corpus embedding generation
18
  @st.cache_data
19
- def generate_embeddings(model, dataset_file, sample_size=32):
20
  # Prepare paper texts by combining query and answer fields
21
  paper_texts = [
22
  record['query'] + '[SEP]' + record['answer'] for record in dataset_file.select(range(sample_size))
23
  ]
24
  # Compute embeddings for all paper texts
25
- return paper_texts, model.encode(paper_texts, convert_to_tensor=True, show_progress_bar=True)
26
 
27
  # Cache summarization pipeline
28
  @st.cache_resource
 
16
 
17
  # Cache corpus embedding generation
18
  @st.cache_data
19
+ def generate_embeddings(_model, dataset_file, sample_size=32):
20
  # Prepare paper texts by combining query and answer fields
21
  paper_texts = [
22
  record['query'] + '[SEP]' + record['answer'] for record in dataset_file.select(range(sample_size))
23
  ]
24
  # Compute embeddings for all paper texts
25
+ return paper_texts, _model.encode(paper_texts, convert_to_tensor=True, show_progress_bar=True)
26
 
27
  # Cache summarization pipeline
28
  @st.cache_resource