final succes

main.py

@@ -101,7 +101,6 @@ async def upload_file(picture: UploadFile):
     os.makedirs(UPLOAD_FOLDER, exist_ok=True)
     file_path = os.path.join(UPLOAD_FOLDER, (picture.filename))
 
-    
 
     with open(file_path, 'wb') as f:
         f.write(picture.file.read())

train_pred.py

@@ -39,9 +39,9 @@ class TrainPred:
         return history
 
     def prediction(self, model):
-        # img_width, img_height = 200, 200
-        # img = load_img("./static/result_upload2.jpg", target_size=(img_width, img_height))
-        img = load_img("./static/result_upload2.jpg")
+        img_width, img_height = 200, 200
+        img = load_img("./static/result_upload2.jpg", target_size=(img_width, img_height))
+        # img = load_img("./static/result_upload2.jpg")
 
         img = img_to_array(img)
         img = img/255.0