Bhadralab commited on
Commit
ef06ce0
·
verified ·
1 Parent(s): afe1d29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -5,10 +5,9 @@ import gradio as gr
5
  from io import StringIO
6
  from transformers import EsmForSequenceClassification, EsmTokenizer
7
 
8
- # Load model and tokenizer once
9
  tokenizer = EsmTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D", do_lower_case=False)
10
  model = EsmForSequenceClassification.from_pretrained("model/best_model5")
11
- model.eval() # Set to evaluation mode for efficiency
12
 
13
  def read_fasta(fasta_string):
14
  """Parses FASTA format input and returns sequences with validation."""
 
5
  from io import StringIO
6
  from transformers import EsmForSequenceClassification, EsmTokenizer
7
 
 
8
  tokenizer = EsmTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D", do_lower_case=False)
9
  model = EsmForSequenceClassification.from_pretrained("model/best_model5")
10
+ model.eval()
11
 
12
  def read_fasta(fasta_string):
13
  """Parses FASTA format input and returns sequences with validation."""