drjieliu commited on
Commit
0b1d749
·
1 Parent(s): 6991d2e

Update func_gradio.py

Browse files
Files changed (1) hide show
  1. func_gradio.py +5 -2
func_gradio.py CHANGED
@@ -6,7 +6,7 @@ from scipy.sparse import load_npz
6
  import matplotlib.pyplot as plt
7
  from matplotlib.gridspec import GridSpec
8
  import matplotlib
9
- def predict_func(input_chrom,cop_type, region_start,region_end, atac_seq,progress=gradio.Progress()):
10
  device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
11
  print(device)
12
  if input_chrom == '' or cop_type == '':
@@ -161,4 +161,7 @@ def make_plots(in_file,md,epis,epi_type, maxv1, maxv2,maxv3):
161
 
162
  axs[-1].set_xlabel('chr%s:%s-%s'%(chrom,start,end))
163
  plt.show()
164
- return fig
 
 
 
 
6
  import matplotlib.pyplot as plt
7
  from matplotlib.gridspec import GridSpec
8
  import matplotlib
9
+ def predict_func(cop_type,input_chrom, region_start,region_end, atac_seq,progress=gradio.Progress()):
10
  device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
11
  print(device)
12
  if input_chrom == '' or cop_type == '':
 
161
 
162
  axs[-1].set_xlabel('chr%s:%s-%s'%(chrom,start,end))
163
  plt.show()
164
+ return fig
165
+
166
+
167
+