Update app.py

app.py

@@ -177,24 +177,20 @@ 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_path = os.path.join(outdir, "submission_det.csv")
-    submission_det.to_csv(submission_det_path, index=False)
-
-    return submission_det_path
+    submission_det.to_csv("submission_det.csv", index=False)
+    return "submission_det.csv"
 
 @app.route('/predict', methods=['POST'])
 def predict():
-    image_id = request.form['image_id']
     image_file = request.files['image']
+    image_id = request.form['image_id']
     model_path = request.form['model_path']
-
-    local_image_path = os.path.join("input_images", image_file.filename)
-    os.makedirs("input_images", exist_ok=True)
+    
+    local_image_path = os.path.join("/tmp", image_file.filename)
     image_file.save(local_image_path)
-
-    submission_det_path = prediction(image_id, local_image_path, model_path)
-
-    return send_file(submission_det_path, as_attachment=True)
+    
+    result_path = prediction(image_id, local_image_path, model_path)
+    return send_file(result_path, as_attachment=True)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=8000)