23 lines
506 B
Python
23 lines
506 B
Python
from search import PaperSearcher
|
|
|
|
# Use local model (free)
|
|
searcher = PaperSearcher('iclr2026_papers.json', model_type='local')
|
|
|
|
# Or use OpenAI (better quality)
|
|
# searcher = PaperSearcher('iclr2026_papers.json', model_type='openai')
|
|
|
|
searcher.compute_embeddings()
|
|
|
|
examples = [
|
|
{
|
|
"title": "Solid-State battery",
|
|
"abstract": "Solid-State battery"
|
|
},
|
|
]
|
|
|
|
results = searcher.search(examples=examples, top_k=100)
|
|
|
|
searcher.display(results, n=10)
|
|
searcher.save(results, 'results.json')
|
|
|