Update README.md

README.md

@@ -8,9 +8,6 @@ model-index:
   results: []
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # titletor-phi_1-5
 
 This model is a fine-tuned version of [microsoft/phi-1_5](https://huggingface.co/microsoft/phi-1_5) on an unknown dataset.
@@ -19,15 +16,24 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+### Sample Code
 
-## Intended uses & limitations
+```python
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-More information needed
+model = AutoModelForCausalLM.from_pretrained("zelalt/titletor-phi_1-5", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("zelalt/titletor-phi_1-5", trust_remote_code=True)
+inputs = tokenizer(f'''What is the title of this paper? ....[your pdf as text]\n\nAnswer: ''', return_tensors="pt", return_attention_mask=False)
+outputs = model.generate(**inputs,max_new_tokens=50, pad_token_id = tokenizer.eos_token_id, eos_token_id = tokenizer.eos_token_id)
+text = tokenizer.batch_decode(outputs)[0]
+print(text)
+```
 
 ## Training and evaluation data
+Train and validation dataset:
+[zelalt/scientific-papers-3.5-withprompt](https://huggingface.co/datasets/zelalt/scientific-papers-3.5-withprompt)
 
-More information needed
 
 ## Training procedure