minor changes

app.py

@@ -7,7 +7,7 @@ import torch.nn as nn
 import cv2
 import numpy as np
 import torch
-import torch.nn as nn
+import torch.nn as nnst
 import torchvision.transforms.functional as TF
 from torchvision import transforms
 
@@ -49,9 +49,16 @@ def blurr_image(input_image,preds):
     return inp,blurred_img
 
 import streamlit as st
-st.title("Image Colorizer")
-        
-file=st.file_uploader("Please upload the B/W image",type=["jpg","jpeg","png"])
+st.title("AI Portrait Mode")
+st.markdown("Creator: [Pranav Kushare] (https://github.com/Pranav082001)")
+# st.markdown(
+#     "Source code: [GitHub Repository](git link)")
+# )
+
+file=st.file_uploader("Please upload the image",type=["jpg","jpeg","png"])
+check=st.checkbox("Dsiplay Mask", value=False)
+
+
 print(file)
 if file is None:
   st.text("Please Upload an image")
@@ -62,7 +69,8 @@ else:
     inp_img,blurred=blurr_image(opencv_image,pred)
     st.text("Original")
     st.image(inp_img)
-    st.text("Mask!!")
-    st.image(pred)
+    if check:
+        st.text("Mask!!")
+        st.image(pred)
     st.text("Blurred")
     st.image(blurred)