Update app.py

app.py

@@ -1,5 +1,3 @@
-# @title Define the application
-
 import time
 import random
 import gradio as gr
@@ -1144,10 +1142,12 @@ def predict(dataset, attributes, access_token):
         predictions_count = len(prediction_results)
         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)
     else:
         prediction_message = "Failed to make predictions."
         prediction_results = pd.DataFrame([])
-
+ 
     return prediction_message, prediction_results
 
 
@@ -1167,9 +1167,6 @@ def load_dataset_and_predict(dataset, attributes, access_token):
 
     return loaded_data, prediction_message, code_example
 
-
-def download_prediction(dataset_preview):
-    return
     
 theme = gr.themes.Default().set(
     loader_color="#505AE7",