Samuel Sledzieski commited on
Commit
d43f920
1 Parent(s): ff2b104

Add error checking for fasta file loading

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -22,7 +22,11 @@ def predict(model, sequence_file, pairs_file):
22
  model = get_pretrained(model_map[model])
23
 
24
  gr.Info("Loading files...")
25
- seqs = SeqIO.to_dict(SeqIO.parse(sequence_file.name, "fasta"))
 
 
 
 
26
  if Path(pairs_file.name).suffix == ".csv":
27
  pairs = pd.read_csv(pairs_file.name)
28
  elif Path(pairs_file.name).suffix == ".tsv":
 
22
  model = get_pretrained(model_map[model])
23
 
24
  gr.Info("Loading files...")
25
+ try:
26
+ seqs = SeqIO.to_dict(SeqIO.parse(sequence_file.name, "fasta"))
27
+ except ValueError as e:
28
+ gr.Error("Invalid FASTA file - duplicate entry")
29
+
30
  if Path(pairs_file.name).suffix == ".csv":
31
  pairs = pd.read_csv(pairs_file.name)
32
  elif Path(pairs_file.name).suffix == ".tsv":