wt002 commited on
Commit
358f8ca
·
verified ·
1 Parent(s): c53e028

Update app.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -8
agent.py CHANGED
@@ -348,15 +348,11 @@ embedding_model = BERTEmbeddings(model_name="bert-base-uncased")
348
 
349
  # Sample text (replace with your own text)
350
  docs = [
351
- Document(page_content="Mercedes Sosa was an Argentine singer and musician."),
352
- Document(page_content="Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"),
353
- Document(page_content="Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?"),
354
- Document(page_content="What is the surname of the equine veterinarian mentioned in 1.E Exercises from the chemistry materials licensed by Marisa Alviar-Agnew & Henry \
355
- Agnew under the CK-12 license in LibreText's Introductory Chemistry materials as compiled 08/21/2023?"),
356
- Document(page_content="What is the final numeric output from the attached Python code?"),
357
- Document(page_content="Who did the actor who played Ray in the Polish-language version of Everybody Loves Raymond play in Magda M.? Give only the first name.")
358
  ]
359
-
360
  # Get the embeddings for the documents
361
  embeddings = embedding_model.embed([doc.page_content for doc in docs])
362
 
 
348
 
349
  # Sample text (replace with your own text)
350
  docs = [
351
+ Document(page_content="Mercedes Sosa released many albums between 2000 and 2009."),
352
+ Document(page_content="She was a prominent Argentine folk singer."),
353
+ Document(page_content="Her album 'Al Despertar' was released in 1998."),
354
+ Document(page_content="She continued releasing music well into the 2000s.")
 
 
 
355
  ]
 
356
  # Get the embeddings for the documents
357
  embeddings = embedding_model.embed([doc.page_content for doc in docs])
358