Upd models urls

README.md

@@ -44,13 +44,21 @@ tags:
 
 
 ### 🧠 **MedSwin Medical Specialist Models**
-- **MedSwin DT** (85% acc) - DaRE-TIES merged model from SFT and KD.
-- **MedSwin NSL** (85% acc) - NuSLERP merged model from SFT and KD.
-- **MedSwin TA** (84% acc) - Task-Arimethic merged model from SFT and KD.
-- **MedSwin SFT** - (82% acc) Supervised Fine-Tuned model.
-- **MedSwin KD** - (83% acc) Knowledge Distillation from MedGemma-27b-it. 
+
+MedLLM Agent supports multiple MedSwin models, each optimized for different use cases:
+
+- **[MedSwin DT](https://huggingface.co/MedSwin/MedSwin-Merged-DaRE-TIES-KD-0.7)** (85% acc) - **Default model**. DaRE-TIES merged model combining Supervised Fine-Tuning (SFT) and Knowledge Distillation (KD) techniques. Best overall performance.
+- **[MedSwin Nsl](https://huggingface.co/MedSwin/MedSwin-Merged-NuSLERP-KD-0.7)** (85% acc) - NuSLERP merged model from SFT and KD. Alternative high-performance option with different merging strategy.
+- **[MedSwin DL](https://huggingface.co/MedSwin/MedSwin-Merged-DaRE-Linear-KD-0.7)** (84% acc) - DaRE-Linear merged model using linear interpolation for knowledge distillation.
+- **[MedSwin Ti](https://huggingface.co/MedSwin/MedSwin-Merged-TIES-KD-0.7)** (84% acc) - TIES merged model from SFT and KD. Compact merged architecture.
+- **[MedSwin TA](https://huggingface.co/MedSwin/MedSwin-Merged-TA-SFT-0.7)** (84% acc) - Task-Arithmetic merged model from SFT. Optimized for task-specific performance.
+- **[MedSwin SFT](https://huggingface.co/MedSwin/MedSwin-7B-SFT)** (82% acc) - Supervised Fine-Tuned model. Base fine-tuned model without merging.
+- **[MedSwin KD](https://huggingface.co/MedSwin/MedSwin-7B-KD)** (83% acc) - Knowledge Distillation model from MedGemma-27b-it. Distilled knowledge from larger model.
+
+**Model Features:**
 - Models download on-demand for efficient resource usage
 - Fine-tuned on MedAlpaca-7B for medical domain expertise
+- All models support GPU acceleration for fast inference
 
 
 ### 🌍 **Multi-Language Support**

ui.py

@@ -256,6 +256,25 @@ def create_demo():
                             value=0.5,
                             label="Merge Threshold (lower = more merging)"
                         )
+                    
+                    # MedSwin Model Links
+                    gr.Markdown(
+                        """
+                        <div style="margin-top: 20px; padding: 15px; background-color: #f5f5f5; border-radius: 8px;">
+                            <h4 style="margin-top: 0; margin-bottom: 10px;">🔗 MedSwin Models on Hugging Face</h4>
+                            <div style="display: flex; flex-wrap: wrap; gap: 10px;">
+                                <a href="https://huggingface.co/MedSwin/MedSwin-Merged-DaRE-TIES-KD-0.7" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin DT</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-Merged-NuSLERP-KD-0.7" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin Nsl</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-Merged-DaRE-Linear-KD-0.7" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin DL</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-Merged-TIES-KD-0.7" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin Ti</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-Merged-TA-SFT-0.7" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin TA</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-7B-SFT" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin SFT</a>
+                                <a href="https://huggingface.co/MedSwin/MedSwin-7B-KD" target="_blank" style="text-decoration: none; padding: 5px 10px; background-color: #1a73e8; color: white; border-radius: 4px; font-size: 12px;">MedSwin KD</a>
+                            </div>
+                            <p style="margin-top: 10px; margin-bottom: 0; font-size: 11px; color: #666;">Click any model name to view details on Hugging Face</p>
+                        </div>
+                        """
+                    )
                 
                 show_thoughts_state = gr.State(value=False)