Update app.py

app.py

@@ -70,7 +70,7 @@ def transalation(img):
     tx=np.random.randint(2,60)
     ty=np.random.randint(2,55)
     tm=np.array([[1,0,tx],[0,1,ty]],dtype=np.float32)
-    trans_img=cv2.warpAffine(img,tm)
+    trans_img=cv2.warpAffine(img,tm,dsize=(300,300))
     img=cv2.imdecode(trans_img,cv2.IMREAD_COLOR)
     img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     st.image(img_rgb, caption="Uploaded Image (OpenCV)", use_container_width=True)