Update app.py

app.py

@@ -232,7 +232,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 🔥 ImageNet ResNet50 Classifier
     
-    **Trained from scratch to 78%+ Top-1 accuracy on ImageNet!**
+    **Trained from scratch to 77%+ Top-1 accuracy on ImageNet!**
     
     Upload any image and get top-5 predictions with confidence scores.
     """)
@@ -258,22 +258,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             - **Training:** From scratch (no pretrained weights)
             - **Dataset:** ImageNet (1.2M images, 1000 classes)
             - **Accuracy:** 77.09% Top-1 validation
-            - **Training Time:** ~13 hours on 8× A100 GPUs
             
             ### 🔗 Links:
-            - [GitHub Repository](https://github.com/Shwethaamrutha/TSAI-S8)
-            - [Training Logs & Details](https://github.com/Shwethaamrutha/TSAI-S8/blob/main/imagenet-training-final/README.md)
-            - [YouTube Demo](https://youtube.com/YOUR_VIDEO_ID)
+            - [GitHub Repository](https://github.com/Shwethaamrutha/TSAI-S9)
             """)
     
     # Example images
     gr.Markdown("### 🖼️ Try These Examples:")
     gr.Examples(
         examples=[
-            ["examples/dog.jpg"],
-            ["examples/cat.jpg"],
-            ["examples/car.jpg"],
-            ["examples/bird.jpg"],
+            ["GermanShephard.jpg"],
+            ["Goldfish.jpg"],
+            ["Tiger.jpg"],
+            ["Eagle.jpg"],
         ],
         inputs=image_input,
         outputs=output,
@@ -284,46 +281,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # Connect button
     predict_btn.click(fn=predict, inputs=image_input, outputs=output)
     
-    gr.Markdown("""
-    ---
-    ### 📊 Training Details:
-    
-    **Phase 1: Initial Training (90 epochs)**
-    - Optimizer: SGD + Nesterov momentum
-    - LR Schedule: OneCycleLR (0.02 → 0.2 → 0.00001)
-    - Regularization: Label smoothing, weight decay, dropout
-    - Result: 76.75%
-    
-    **Phase 2: Fine-tuning (Multiple LR restarts)**
-    - LR=0.001: 76.88% (oscillated)
-    - LR=0.0005: **77.09%** ✅ (best achieved!)
-    - LR=0.0003: 77.02% (similar ceiling)
-    
-    **Result:** 77.09% represents the natural ceiling for standard
-    from-scratch training. Achieving 78%+ requires advanced augmentation
-    techniques (MixUp, CutMix) beyond standard methods.
-    
-    **Key Techniques:**
-    - Mixed precision training (torch.amp)
-    - Distributed training (8 GPUs, DDP)
-    - Robust image loading (handles corrupted files)
-    - Advanced augmentation (crop, flip, color jitter, erasing)
-    
-    ### 💰 Cost Analysis:
-    - Hardware: AWS p4d.24xlarge (8× A100 40GB)
-    - Duration: ~13 hours
-    - Cost: ~$110 (spot pricing)
-    
-    ### 📊 Performance Context:
-    - **Industry Baseline:** 70-75% (we beat by 2-7%)
-    - **Good Training:** 75-77% (top tier!)
-    - **Our Result:** 77.09% (top 10% of from-scratch)
-    - **Research-Level:** 78%+ (requires MixUp/CutMix)
-    
-    ---
-    
-    **Made with ❤️ by Shwetha(https://github.com/Shwethaamrutha)**
-    """)
+
 
 # Launch
 if __name__ == "__main__":