Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,7 +10,6 @@ import torchvision.transforms as T
 import matplotlib.pyplot as plt
 from huggingface_hub import hf_hub_download
 
-# ========== Model Definition ==========
 class MobileViTSegmentation(nn.Module):
     def __init__(self, encoder_name='mobilevit_s', pretrained=True):
         super().__init__()