Update README.md

README.md

@@ -1,3 +1,15 @@
+---
+datasets:
+- Omartificial-Intelligence-Space/Arabic-finanical-rag-embedding-dataset
+language:
+- ar
+base_model:
+- ybelkada/falcon-7b-sharded-bf16
+pipeline_tag: text-generation
+library_name: transformers
+tags:
+- finance
+---
 # Model: FalconMasr
 
 This model is based on the Falcon-7B model with quantization in 4-bit format for efficient memory usage and fine-tuned using LoRA (Low-Rank Adaptation) for Arabic causal language modeling tasks. The model has been configured to handle causal language modeling tasks specifically designed to improve responses in Arabic.
@@ -111,4 +123,4 @@ output = model.generate(**inputs, max_length=200,
 
 # Decode the generated output
 decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
-print(decoded_output)
+print(decoded_output)