Update app.py

app.py

@@ -11,9 +11,8 @@ print('Modules installed')
 os.system('pip install --force gradio==3.36.1')
 os.system('pip install gradio_client==0.2.7')
 #os.system('pip install numpy --upgrade')
-os.system('pip install --force numpy==1.24.1')
+#os.system('pip install --force numpy==1.24.1')
 
-os.environ["DGLBACKEND"] = "pytorch"
 
 if not os.path.exists('./SEQDIFF_230205_dssp_hotspots_25mask_EQtasks_mod30.pt'):
     print('Downloading model weights 1')
@@ -216,6 +215,7 @@ def protein_diffusion_model(sequence, seq_len, helix_bias, strand_bias, loop_bia
     # sampling loop
     plddt_data = []
     for j in range(S.max_t):
+        print(f'on step {j}')
         output_seq, output_pdb, plddt = S.take_step_get_outputs(j)
         plddt_data.append(plddt)
         yield output_seq, output_pdb, display_pdb(output_pdb), get_plddt_plot(plddt_data, S.max_t)