Update app.py

app.py

@@ -85,10 +85,13 @@ log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLAS
 model_weights_path = '/gena-blastln-bs33-lr4e-05-S168.pth'
 weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
 
-model = BertClassifier(base_model, log_reg, num_labels = N_UNIQUE_CLASSES)
 base_model.load_state_dict(weights['model_state_dict'])
 log_reg.load_state_dict(weights['log_reg_state_dict'])
 
+# Creating Model
+model = BertClassifier(base_model, log_reg, num_labels = N_UNIQUE_CLASSES)
+
+
 # Define a function to process the DNA sequence
 def analyze_dna(sequence):
     # Preprocess the input sequence