Update README.md

README.md

@@ -1,16 +1,15 @@
 # ArtifyAI v1.0: Text-to-Image Generation
 
-ArtifyAI v1.0 is a project designed to generate images from text using the T5 model for text summarization and the Stable Diffusion model for image generation. This version also provides functionality to save model weights and load them for future use, making it easier to resume your work.
+ArtifyAI v1.0 is a project designed to generate images using the Stable Diffusion model. This version focuses on saving and loading model weights, specifically for the image generation pipeline. It also provides functionality to store and retrieve models using Google Drive.
 
 ## Overview
 
-ArtifyAI v1.0 uses T5 for text processing, Stable Diffusion for image generation, and includes functionality for saving and loading the model's weights. You can easily store model weights in Google Drive and use them later without re-downloading everything.
+ArtifyAI v1.0 uses Stable Diffusion for high-quality image generation based on descriptive inputs. The project allows saving and loading of model weights to/from Google Drive, so you don't need to re-download models in future sessions.
 
 ## Features
 
-- **Text Processing**: Utilizes T5 for text summarization and generation.
-- **Image Generation**: Uses Stable Diffusion for high-quality image generation based on textual descriptions.
-- **Model Saving**: Allows you to save and reload model weights (e.g., UNet) to/from Google Drive.
+- **Image Generation**: Uses Stable Diffusion to generate images from text descriptions.
+- **Model Saving**: Supports saving and loading of model weights (e.g., `UNet`) to/from Google Drive.
 
 ## Installation
 
@@ -20,7 +19,7 @@ Ensure that you have the following to run this project:
 
 1. Python 3.7+
 2. CUDA-compatible GPU (for faster performance)
-3. [Hugging Face Transformers](https://huggingface.co/transformers/)
+3. [Hugging Face Transformers](https://huggingface.co/transformers/) library
 4. [Diffusers](https://huggingface.co/docs/diffusers/index)
 5. [PyTorch](https://pytorch.org/) with CUDA support
 
@@ -37,16 +36,13 @@ Ensure that you have the following to run this project:
     pip install torch transformers diffusers
     ```
 
-3. **Download the Pretrained Models**:
-    Use the following code to load the models:
+3. **Download the Pretrained Model**:
+    Use the following code to load the Stable Diffusion model:
     ```python
-    from transformers import T5Tokenizer, T5ForConditionalGeneration
     from diffusers import StableDiffusionPipeline
     import torch
 
-    # Load models
-    t5_tokenizer = T5Tokenizer.from_pretrained("t5-small")
-    t5_model = T5ForConditionalGeneration.from_pretrained("t5-small")
+    # Load the Stable Diffusion model
     ArtifyAI_model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
 
     # Set model to GPU if available
@@ -99,4 +95,3 @@ If you're not familiar with coding or AI, follow these steps to use ArtifyAI:
 3. **Run the Project**: Use the code snippets provided in the notebook or above to generate images from text.
 
 4. **Save Your Work**: If you are using Google Colab, remember to save your models to Google Drive to keep your work.
-