Update app.py

app.py

@@ -177,20 +177,24 @@ def prediction(image_id_main, local_image_path, model_path):
             results_list.append(result)
 
     submission_det = pd.DataFrame(results_list, columns=['image_id', 'PredictionString'])
-    submission_det.to_csv("submission_det.csv", index=False)
-    return "submission_det.csv"
+    submission_det_path = os.path.join(outdir, "submission_det.csv")
+    submission_det.to_csv(submission_det_path, index=False)
+
+    return submission_det_path
 
 @app.route('/predict', methods=['POST'])
 def predict():
-    image_file = request.files['image']
     image_id = request.form['image_id']
+    image_file = request.files['image']
     model_path = request.form['model_path']
-    
-    local_image_path = os.path.join("/tmp", image_file.filename)
+
+    local_image_path = os.path.join("input_images", image_file.filename)
+    os.makedirs("input_images", exist_ok=True)
     image_file.save(local_image_path)
-    
-    result_path = prediction(image_id, local_image_path, model_path)
-    return send_file(result_path, as_attachment=True)
+
+    submission_det_path = prediction(image_id, local_image_path, model_path)
+
+    return send_file(submission_det_path, as_attachment=True)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=8000)