Update app.py

app.py

@@ -2,7 +2,8 @@ import os
 import zipfile
 import numpy as np
 import torch
-from transformers import SegformerForImageSegmentation, ResNetForImageClassification, AdamW
+from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtractor
+from transformers import ResNetForImageClassification, AdamW
 from PIL import Image
 from torch.utils.data import Dataset, DataLoader
 import streamlit as st