Shinapri commited on
Commit
5969ef9
verified
1 Parent(s): 6853496

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -6
README.md CHANGED
@@ -56,10 +56,11 @@ model_id = "Shinapri/Matellem-Gemma3n-E2B-Graphene-1"
56
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto").eval()
57
  tokenizer = AutoTokenizer.from_pretrained(model_id)
58
 
59
- INSTRUCTION = "You are a scientific literature search expert. Your task is to identify the title of a research paper based on a user's description of its key methods and findings."
60
- USER_INPUT = """I'm looking for a paper about manipulating graphene plasmons.
61
- The key method involved using a ferroelectric nanocavity array to create a periodic doping pattern on the graphene.
62
- I remember they could tune the plasmon resonance by dynamically changing the applied gate voltage. Can you identify the title?"""
 
63
 
64
  messages = [
65
  {"role": "system", "content": [{"type": "text", "text": INSTRUCTION}]},
@@ -86,9 +87,9 @@ with torch.no_grad():
86
 
87
  The model's response for this query is highly relevant:
88
 
89
- > The title of the paper is likely: **"Voltage-tunable plasmonics on few-layer graphene based on a ferroelectric nanocavity array"**
90
 
91
- For reference, the original paper's title is **"Tunable plasmonic devices by integrating graphene with ferroelectric nanocavity"**. A cosine similarity score of **0.9638** was obtained when comparing the embeddings of these two titles using the `intfloat/e5-large` model.
92
  -----
93
 
94
  ## Future Work / Roadmap
 
56
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto").eval()
57
  tokenizer = AutoTokenizer.from_pretrained(model_id)
58
 
59
+ INSTRUCTION = "You are a scientific literature search expert. Your task is to suggest a likely title for a research paper based on the user's description."
60
+ USER_INPUT = """I'm trying to recall the title of a review paper I read.
61
+ It discussed the applications of graphene specifically in the field of neuroscience,
62
+ covering two main areas: its use as a carrier for drug delivery and as a conductive substrate for tissue engineering.
63
+ Can you suggest a likely title?"""
64
 
65
  messages = [
66
  {"role": "system", "content": [{"type": "text", "text": INSTRUCTION}]},
 
87
 
88
  The model's response for this query is highly relevant:
89
 
90
+ > Based on your description, a likely title for the review paper is: **Graphene in Neuroscience: Applications in Drug Delivery and Tissue Engineering**
91
 
92
+ For reference, the original paper's title is **"Interfacing Graphene-Based Materials With Neural Cells"**. A cosine similarity score of **0.9396** was obtained when comparing the embeddings of these two titles using the `intfloat/e5-large` model.
93
  -----
94
 
95
  ## Future Work / Roadmap