Update app.py

app.py

@@ -63,22 +63,21 @@ def transalation(img):
     tx=3 #np.random.randint(2,60)
     ty=5 #np.random.randint(2,55)
     tm=np.array([[1,0,tx],[0,1,ty]],dtype=np.float32)
-    trans_img=cv2.warpAffine(img,tm,dsize=(20,20))
-    img=cv2.imdecode(trans_img,cv2.IMREAD_COLOR)
+    trans_img=cv2.warpAffine(img,tm,dsize=(img.shape[1],img.shape[0]))
     img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    st.image(img_rgb, caption="Uploaded Image (OpenCV)", use_container_width=True)
+    st.image(img_rgb, caption="Translation Applied", use_container_width=True)
 
 
 if uploaded_file is not None:
     # Converting the image into array
     img_array=np.asarray(bytearray(uploaded_file.read()),dtype=np.uint8)
-    # Decode or converting the array back to image
-    # img=cv2.imdecode(img_array,cv2.IMREAD_COLOR)
+    # Decode or converting the array back to image array
+    img=cv2.imdecode(img_array,cv2.IMREAD_COLOR)
     #  # Convert BGR to RGB as streamlit expects RGB colorspace
     # img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     transformation = []
     if trans:
-        transalation(img_array)
+        transalation(img)
         transformation.append("Translation")
     if rot:
         transformation.append("Rotation")