Update README.md

README.md

@@ -1,3 +1,33 @@
+---
+library_name: transformers
+tags: []
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
+
+- **Developed by:** Sefika Efeoglu
+
+- **Model type:** text-to-text
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** https://huggingface.co/google/flan-t5-base
+
+
+## Uses
+
+```python
 import json
 import torch
 from transformers import AutoTokenizer
@@ -20,4 +50,34 @@ inputs = self.tokenizer(prompt, add_special_tokens=True, max_length=526,return_t
 outputs = self.model.generate(inputs, max_new_tokens=length,  pad_token_id=self.tokenizer.eos_token_id)
 
 response = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
-print(response[0])
+print(response[0])
+#"Cause-Effect(e1,e2)"
+
+```
+
+
+## Training Details
+
+### Training Data
+
+semeval-2010-task8
+[More Information Needed]
+
+### Training Procedure
+
+5 fold cross validation with sentence and relation types. Input is sentence and the output is relation types
+
+#### Training Hyperparameters
+
+Epoch:5, BS:16 and others are default.
+
+#### Hardware
+
+Colab Pro+ A100.
+
+
+
+## Citation [optional]
+
+Efeoglu, Sefika, and Adrian Paschke. "Retrieval-Augmented Generation-based Relation Extraction." arXiv preprint arXiv:2404.13397 (2024).
+