Update app.py

app.py

@@ -220,18 +220,10 @@ def ACE(file):
     test_dict = {"text":test_sequences, 'structure':test_Structure_index}
     print("=================================Structure prediction========================")
     for i in tqdm(range(0, len(test_sequences))):
-        while True:
-            command = ["curl", "-X", "POST", "-k", "--data", f"{test_sequences[i]}", "https://api.esmatlas.com/foldSequence/v1/pdb/"]
-            result = subprocess.run(command, capture_output=True, text=True)
-            with open(os.path.join(pdb_path, f'{test_Structure_index[i]}.pdb'), 'w') as file:
-                file.write(result.stdout)
-            stats = os.stat(os.path.join(pdb_path, f'{test_Structure_index[i]}.pdb'))
-            if stats.st_size < 1024:
-                print(f"Download for {test_Structure_index[i]} failed due to empty file. Retrying...")
-                time.sleep(20)
-                continue
-            else:
-                break
+        command = ["curl", "-X", "POST", "-k", "--data", f"{test_sequences[i]}", "https://api.esmatlas.com/foldSequence/v1/pdb/"]
+        result = subprocess.run(command, capture_output=True, text=True)
+        with open(os.path.join(pdb_path, f'{test_Structure_index[i]}.pdb'), 'w') as file:
+            file.write(result.stdout)
     test_data=MyDataset(test_dict)
     test_dataloader=DataLoader(test_data,batch_size=batch_size,collate_fn=collate_fn,shuffle=False)
 
@@ -250,6 +242,8 @@ def ACE(file):
             batchs = {k: v for k, v in batch.items()}
             outputs = model(structure_fea, batchs, fingerprint)
             probability = outputs[0].tolist()
+            print(outputs)
+            print(probability)
             train_argmax = np.argmax(outputs.cpu().detach().numpy(), axis=1)
             for j in range(0,len(train_argmax)):
                 output = train_argmax[j]