Docs: Updated documentation

README.md

@@ -41,18 +41,24 @@ The model is trained using cross-entropy loss and AdamW optimizer with weight de
 └── requirements.txt    # Dependencies
 ```
 
+
+
 ### Install Dependencies
 
 ```
 pip install -r requirements.txt
 ```
 
+
+
 ### Run the Notebook
 
 ```
 jupyter notebook S12Trained.ipynb
 ```
 
+
+
 ### Training Logs
 
 Training logs for few steps are shown below:
@@ -114,6 +120,8 @@ Reached target loss! Final loss: 0.0889 at step 11,663
 Model saved to gpt_model.pt
 ```
 
+
+
 ### Model Output
 
 ```bash
@@ -126,6 +134,11 @@ The tyrant from one curst my death lies;
 For the ground is nothing henceforth fell executioner come
 ```
 
+
+
 ### Try it out
 
-App Link
+App Link: https://huggingface.co/spaces/Shilpaj/ShakespeareGPT
+
+
+![App](./assets/app.gif)

app.py

@@ -159,8 +159,8 @@ demo = gr.Interface(
         gr.Slider(minimum=0.1, maximum=2.0, value=0.8, step=0.1, label="Temperature (higher = more random)")
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="NanoGPT Text Generator",
-    description="Generate Shakespeare-style text using a trained NanoGPT model",
+    title="GPT Text Generator",
+    description="Generate Shakespeare-style text using a trained GPT model",
     allow_flagging="never",
     cache_examples=True
 )