Update app.py

app.py

@@ -1143,7 +1143,7 @@ def predict(dataset, attributes, access_token):
         accuracy = random.uniform(0.85, 0.95)
         prediction_message = f"{predictions_count} predictions made in {elapsed_time:.2f} seconds with an accuracy of {accuracy:.2f}%"
 
-        prediction_results.to_csv("/file/prediction.csv", index=False)
+        prediction_results.to_csv("prediction.csv", index=False)
     else:
         prediction_message = "Failed to make predictions."
         prediction_results = pd.DataFrame([])
@@ -1208,7 +1208,7 @@ with gr.Blocks(theme=theme) as demo:
         code_example = gr.Code(language="python")
 
     predict_button = gr.Button("Predict Attributes")
-    download_button = gr.DownloadButton(label=f"Download Prediction", value="/file/prediction.csv")
+    download_button = gr.DownloadButton(label=f"Download Prediction", value="/prediction.csv")
     
     selected_dataset.change(
         fn=load_dataset,
@@ -1234,4 +1234,4 @@ with gr.Blocks(theme=theme) as demo:
         outputs=[dataset_preview, prediction_label, code_example],
     )
 
-demo.launch(allowed_paths=["/file/prediction.csv"])
+demo.launch(allowed_paths=["/prediction.csv"])