Update app.py

app.py

@@ -20,11 +20,11 @@ def inference():
     image = cv2.imread(secure_filename(file_name))
     w,h = image.shape[1],image.shape[0]
     image = cv2.resize(image,(int(w*.79),int(h*.79)))
-    cv2.imwrite(secure_filename(file.filename),image)
+    cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.jpg',image)
     os.system("uname -a")
     #output = remove(image)  # remove background
     outputName = secure_filename(file.filename).split('.')[0]+'.png'
-    os.system('backgroundremover -i'+'{name}'.format(name= secure_filename(file.filename)) + ' -m "u2net_human_seg" -o {name}'.format(name= outputName))
+    os.system('backgroundremover -i'+'{name}'.format(name= secure_filename(file.filename).split('.')[0]+'.jpg') + ' -m "u2net_human_seg" -o {name}'.format(name= outputName))
 
     print(secure_filename(file.filename).split('.'))
     os.system("ls -l")