Update app.py

app.py

@@ -19,13 +19,13 @@ model.to('cuda')
 def generate_caption(protein, prompt):
     # Process the image and the prompt
     print(f"system path: {os.getcwd()}")
-    with open('data/fasta/example.fasta', 'w') as f:
+    with open('/home/user/app/example.fasta', 'w') as f:
         f.write('>{}\n'.format("protein_name"))
         f.write('{}\n'.format(protein.strip()))
     print(f"fasta prepared")
-    os.system("python esm_scripts/extract.py esm2_t36_3B_UR50D data/fasta/example.fasta data/emb_esm2_3b --repr_layers 36 --truncation_seq_length 1024 --include per_tok")
+    os.system("python esm_scripts/extract.py esm2_t36_3B_UR50D /home/user/app/example.fasta /home/user/app --repr_layers 36 --truncation_seq_length 1024 --include per_tok")
     print(f"protein pt file prepared")
-    esm_emb = torch.load("data/emb_esm2_3b/protein_name.pt")['representations'][36]
+    esm_emb = torch.load("/home/user/app/protein_name.pt")['representations'][36]
     esm_emb = F.pad(esm_emb.t(), (0, 1024 - len(esm_emb))).t().to('cuda')
     samples = {'name': ['test_protein'],
                'image': torch.unsqueeze(esm_emb, dim=0),