Update app.py

app.py

@@ -39,6 +39,13 @@ if uploaded_file is not None:
     img = np.array(img)
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     
+    cust_transform = A.Compose([A.Resize(height=256, width=256, p=1.0),ToTensorV2(p=1.0)], p=1.0)
+    tensor = cust_transform(image=img)
+    tensor = tensor['image'].float(),resize(1,3,256,256)
+    
+    custom_pred = model.forward(tensor).detach().numpy()
+    custom_pred
+    
   elif ".csv" in uploaded_file.name:
     dataframe = pd.read_csv(uploaded_file)
     st.write(dataframe)