Update app.py

app.py

@@ -89,16 +89,18 @@ for img_path in img_list:
                 save_restore_path = os.path.join("_output", 'restored_imgs', f'{basename}.{extension}')
             imwrite(restored_img, save_restore_path)
 os.system("ls ./_output")
+os.system("echo ----")
 os.system("ls ./_output/cmp")
+os.system("echo ----")
 os.system("ls ./_output/restored_imgs")
-
+os.system("echo ----")
 
 
 
 
 def inference():
-    
-    input_img = cv2.imread("./_output/cmp/imagem-0001_0000.png" , cv2.IMREAD_COLOR)
+    random.randint(0, 9)
+    input_img = cv2.imread("./_output/cmp/imagem-000"+str(random.randint(1, 4))+"_0000.png" , cv2.IMREAD_COLOR)
     input_img= cv2.cvtColor(input_img,cv2.COLOR_BGR2RGB) 
     st.image(input_img)