update model code

app.py

@@ -23,6 +23,8 @@ GDRIVE_MODEL_URL = "https://drive.google.com/uc?id=1fzKneepaRt_--dzamTcDBM-9d3_d
 LOCAL_MODEL_PATH = "checkpoint32.pth"
 
 print(GDRIVE_MODEL_URL)
+
+
 def download_file_from_google_drive():
     gdown.download(GDRIVE_MODEL_URL, LOCAL_MODEL_PATH, quiet=False)
 
@@ -227,7 +229,7 @@ def predict():
     if file.filename.lower().endswith((".png", ".jpg", ".jpeg")):
         image = Image.open(temp_path)
         image_save_path = os.path.join(
-            tempfile.gettempdir(), "saved_image.jpg")
+            tempfile.gettempdir(), file.filename.lower())
         image.save(image_save_path)
 
     return jsonify({"message": model.check_file(temp_path), "saved_path": image_save_path})