Update app.py

app.py

@@ -85,7 +85,7 @@ def load_model(model_name: str):
         return model, None
 
     else: 
-        return {"error": "Invalid model name"}
+        raise ValueError("Invalid model name")
         
 
 
@@ -112,24 +112,29 @@ def analyze_dna(username, password, sequence, model_name):
         model, tokenizer = load_model(model_name)
 
         def get_logits(seq, model_name):
+            
             if model_name == 'gena-bert':
+                
                 inputs = tokenizer(seq, truncation=True, padding='max_length', max_length=512, return_tensors="pt", return_token_type_ids=False)
                 with torch.no_grad():
                     logits = model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
                 return logits
 
             elif model_name == 'CNN':
-                # Truncate sequence
+                
                 SEQUENCE_LENGTH = 8000
+                pad_char = 'N'
+                
+                # Truncate sequence
                 seq = seq[:SEQUENCE_LENGTH]
 
                 # Pad sequences to the desired length
-                seq = seq.ljust(length, pad_char)[:SEQUENCE_LENGTH]
+                seq = seq.ljust(SEQUENCE_LENGTH, pad_char)[:SEQUENCE_LENGTH]
 
-                # Apply one-hot encoding to the 'sequence' column
-                input = seq.one_hot_encode()
+                # Apply one-hot encoding to the sequence
+                input_tensor = one_hot_encode(seq).unsqueeze(0)
                 with torch.no_grad():
-                    logits = model(input)
+                    logits = model(input_tensor)
                 return logits