Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,11 @@ import gradio as gr
|
|
2 |
import os
|
3 |
import subprocess
|
4 |
from transformers import AutoModelForSequenceClassification,AutoTokenizer
|
|
|
5 |
model_names = ['plant-dnabert','plant-nucleotide-transformer','plant-dnagpt',
|
6 |
'plant-dnagemma','dnabert2','nucleotide-transformer-v2-100m','agront-1b']
|
|
|
|
|
7 |
|
8 |
def inference(seq,model,task):
|
9 |
if not seq:
|
|
|
2 |
import os
|
3 |
import subprocess
|
4 |
from transformers import AutoModelForSequenceClassification,AutoTokenizer
|
5 |
+
|
6 |
model_names = ['plant-dnabert','plant-nucleotide-transformer','plant-dnagpt',
|
7 |
'plant-dnagemma','dnabert2','nucleotide-transformer-v2-100m','agront-1b']
|
8 |
+
tokenizer_type = "BPE"
|
9 |
+
model_names = [x + '-' + tokenizer_type if x.startswith("plant") else x for x in model_names]
|
10 |
|
11 |
def inference(seq,model,task):
|
12 |
if not seq:
|