Update README.md
Browse files
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
|
60 |
-
USER_INPUT = """I'm
|
61 |
-
|
62 |
-
|
|
|
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 |
-
>
|
90 |
|
91 |
-
For reference, the original paper's title is **"
|
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
|