Update app.py

app.py

@@ -3,19 +3,15 @@ import streamlit as st
 from io import StringIO
 from PIL import Image
 import pandas as pd
-
+import numpy as np
 import torch
 import torch.nn as nn
 import torchvision.models as models
-
 import albumentations as A
 from albumentations.pytorch.transforms import ToTensorV2
-
-import numpy as np
-
 import cv2
 
-st.title('Dummy')
+st.title('Predict Images')
 uploaded_file = st.file_uploader('Select File')
 
 id2class = {0: 'agricultural', 1: 'airplane', 2: 'baseballdiamond', 3: 'beach', 4: 'buildings', 5: 'chaparral', 6: 'denseresidential', 7: 'forest', 8: 'freeway', 9: 'golfcourse', 10: 'intersection', 11: 'mediumresidential', 12: 'mobilehomepark', 13: 'overpass', 14: 'parkinglot', 15: 'river', 16: 'runway', 17: 'sparseresidential', 18: 'storagetanks', 19: 'tenniscourt', 20: 'harbor'}