Model files

app.py

@@ -22,11 +22,11 @@ from GraphAnalysis import GraphAnalysis
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the autoencoder model
-autoencoder_config = OmicsConfig.from_pretrained("/workspace/HNSCC-MultiOmics-Graph-Attention-Autoencoder")
+autoencoder_config = OmicsConfig.from_pretrained("./lc_models/MultiOmicsAutoencoder/trained_autoencoder")
 autoencoder_model = MultiOmicsGraphAttentionAutoencoderModel(autoencoder_config).to(device)
 
 # Initialize Attention Extracter
-graph_data_dict_path = '/workspace/MultiOmics-Graph-Attention-Autoencoder/data/graph_data_dictN.pth'
+graph_data_dict_path = './data/hnscc.patient.chg.network.pth'
 extracter = Attention_Extracter(graph_data_dict_path, autoencoder_model.encoder, gpu=(device == "cuda"))
 
 def extract_features():

data/hnscc.patient.chg.network.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01b4465e33695fa8cc042ef305ba4646df4ae35f57a14d10a2c9b36604129e72
+size 154185872