upload

README.md

@@ -0,0 +1,62 @@
+
+# Stable Diffusion Prompts Generation Model
+
+This Hugging Face model, named "Falah/stable_diffusion_prompts_gen", is designed for generating illustration art style prompts using the Stable Diffusion tool for text-to-image generation. It utilizes the custom dataset "Falah/stable_diffusion_prompts_dataset" to generate creative and coherent text prompts.
+
+## Examples
+
+To load the model and generate inferences using the model, you can use the following code snippet:
+
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+model_name = "Falah/stable_diffusion_prompts_gen"
+dataset_name = "Falah/stable_diffusion_prompts_dataset"
+prompt = r'a beautiful female'  # the beginning of the prompt
+temperature = 0.9              # a higher temperature will produce more diverse results, but with a higher risk of less coherent text
+top_k = 8                      # the number of tokens to sample from at each step
+max_length = 100               # the maximum number of tokens for the output of the model
+repetition_penalty = 1.2       # the penalty value for each repetition of a token
+num_return_sequences = 5       # the number of results to generate
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+input_ids = tokenizer(prompt, return_tensors='pt').input_ids
+output = model.generate(
+    input_ids,
+    do_sample=True,
+    temperature=temperature,
+    top_k=top_k,
+    max_length=max_length,
+    num_return_sequences=num_return_sequences,
+    repetition_penalty=repetition_penalty,
+    early_stopping=True
+)
+
+print('\033[96m' + prompt + '\033[0m')
+for i in range(len(output)):
+    print(tokenizer.decode(output[i], skip_special_tokens=True) + '\n')
+```
+
+Feel free to modify the parameters like `prompt`, `temperature`, `top_k`, etc., to experiment with different outputs.
+
+## Citation
+
+If you use this model or the associated dataset in your research or projects, please cite it as follows:
+
+```
+@article{your-article-citation,
+  title = {Title of Your Article},
+  author = {Your Name},
+  journal = {Journal Name},
+  year = {2023},
+  doi = {10.xxxx/your-doi}
+}
+```
+
+## License
+
+This project is licensed under the Apache License, Version 2.0. Please see the [LICENSE](link-to-license-file) file for more details.
+