Update app.py

app.py

@@ -5,11 +5,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 # Load the Hugging Face model and tokenizer
 model_name = 'AIRI-Institute/gena-lm-bert-base-lastln-t2t'  # Replace with the actual model name
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels = 38)
-
-# Ensure the model has the correct number of classes
-num_classes = model.config.num_labels
-assert num_classes == 38, f"The model has {num_classes} classes, but 38 were expected."
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 # Define a function to process the DNA sequence
 def analyze_dna(sequence):