Update app.py

app.py

@@ -40,8 +40,8 @@ css="""
 .img_box{
     display: flex;
     flex-direction: column;
+    flex-flow: unset;
     flex-wrap: wrap;
-    height: 20000px;
 }
 .img_class{
     background: #ffffff;