fix data download bug

run_prothgt_app.py

@@ -138,16 +138,14 @@ def generate_prediction_df(protein_ids, model_paths, model_config_paths, go_cate
     output = "data/prothgt-kg.pt"
 
     if not os.path.exists(output):
-        url = f"https://drive.google.com/uc?id={file_id}"
-        print(f"Downloading file from {url}...")
+        try:
+            url = f"https://drive.google.com/uc?id={file_id}"
+            print(f"Downloading file from {url}...")
+        except Exception as e:
+            print(f"Error downloading file: {e}")
+            raise
     else:
         print(f"File already exists at {output}")
-    try:
-        gdown.download(url, output, quiet=False)
-        print(f"File downloaded to {output}")
-    except Exception as e:
-        print(f"Error downloading file: {e}")
-        raise
 
     heterodata = torch.load(output)
     print(heterodata.edge_types)