Update README.md

README.md

@@ -8,6 +8,8 @@ metrics:
 model-index:
 - name: text-pic-request-identifier
   results: []
+datasets:
+- andriadze/pic-text-requests-synth
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,24 +17,28 @@ should probably proofread and complete it, then remove this comment. -->
 
 # text-pic-request-identifier
 
-This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an synthetic dataset.
+
+
 It achieves the following results on the evaluation set:
 - Loss: 0.0015
 - Accuracy: 0.9996
 
 ## Model description
 
-More information needed
+Model identifies if user is asking for a picture or a text.
 
 ## Intended uses & limitations
 
-More information needed
+Intended use for chat applications to either route the message to a text model or an image model.
+
+Model will return 'pic' or 'text'
+
 
 ## Training and evaluation data
 
-More information needed
+Model was trained on synthetic dataset consisting of around ~25k messages. Messages were generated by different LLM's including gpt4,gpto,gpt4o-mini,gpt3.5-turbo
 
-## Training procedure
 
 ### Training hyperparameters
 
@@ -58,4 +64,4 @@ The following hyperparameters were used during training:
 - Transformers 4.44.0
 - Pytorch 2.3.1
 - Datasets 2.21.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1