update download

app.py

@@ -31,7 +31,7 @@ class PDBSearchAssistant:
             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
-            device="cuda" # cuda or cpu
+            device="cpu" # cuda or cpu
         )
         
         self.prompt_template = """
@@ -1229,12 +1229,20 @@ def server(input, output, session):
     @output
     @render.download(filename="pdb_search_results.csv")
     def download():
+        file_path = "pdb_search_results.csv"
+        if os.path.exists(file_path):
+            os.remove(file_path)
+
         current_results = results_store.get()
         if current_results["type"] == "structure":
             df = pd.DataFrame(current_results["results"])
         else:
+            print()
             df = pd.DataFrame(current_results["results"])
-        return df.to_csv(index=False)
+
+        df.to_csv(file_path, index=False)
+
+        return file_path
 
 app = App(app_ui, server)