Add error checking for fasta file loading
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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":
|