Update func_gradio.py

func_gradio.py

@@ -6,7 +6,7 @@ from scipy.sparse import load_npz
 import matplotlib.pyplot as plt
 from matplotlib.gridspec import GridSpec
 import matplotlib
-def predict_func(input_chrom,cop_type, region_start,region_end, atac_seq):
+def predict_func(input_chrom,cop_type, region_start,region_end, atac_seq,progress=gradio.Progress()):
     device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
     print(device)
     if input_chrom == '' or cop_type == '':
@@ -29,13 +29,15 @@ def predict_func(input_chrom,cop_type, region_start,region_end, atac_seq):
     else:
         chrom, start, end = check_region(input_chrom, region_start,region_end, ref_genome,1000000)
 
-
-
+    progress(0.4, desc="predicting epigenome ...")
     out_epi_binding = predict_epb(os.path.abspath('models/epi_bind.pt'), [start, end], ref_genome, atac_seq, device,
                                   cop_type)
+    out_epi = predict_epis(os.path.abspath('models/epi_track.pt'), [start, end], ref_genome, atac_seq, device, cop_type)
+
+    progress(0.6, desc="predicting transcriptome ...")
     out_cage = predict_cage(os.path.abspath('models/cage.pt'), [start, end], ref_genome, atac_seq, device, cop_type)
 
-    out_epi = predict_epis(os.path.abspath('models/epi_track.pt'), [start, end], ref_genome, atac_seq, device, cop_type)
+
 
     file_id = str(uuid.uuid4())
 
@@ -45,7 +47,7 @@ def predict_func(input_chrom,cop_type, region_start,region_end, atac_seq):
         for f in os.listdir('results/'):
             os.remove(os.path.join('results/', f))
 
-
+    progress(0.8, desc="predicting chromatin organization ...")
     if cop_type == 'Micro-C':
         out_cop = predict_microc(os.path.abspath('models/microc.pt'), [start, end], ref_genome, atac_seq, device)
         np.savez_compressed( 'results/prediction_%s.npz'%file_id,
@@ -159,7 +161,4 @@ def make_plots(in_file,md,epis,epi_type, maxv1, maxv2,maxv3):
 
     axs[-1].set_xlabel('chr%s:%s-%s'%(chrom,start,end))
     plt.show()
-    return fig
-
-
-
+    return fig