lgq12697 commited on
Commit
91f37c6
·
verified ·
1 Parent(s): 421c348

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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: