Update app.py

app.py

@@ -146,7 +146,7 @@ def analyze_dna(username, password, sequence, model_name):
 
         def get_logits(seq, model_name):
             
-            if model_name == 'gena-bert':
+            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():
@@ -165,10 +165,13 @@ def analyze_dna(username, password, sequence, model_name):
                 seq = seq.ljust(SEQUENCE_LENGTH, pad_char)[:SEQUENCE_LENGTH]
 
                 # Apply one-hot encoding to the sequence
-                input_tensor = one_hot_encode(seq).unsqueeze(0)
+                input_tensor = one_hot_encode(seq).unsqueeze(0).float() 
                 with torch.no_grad():
                     logits = model(input_tensor)
                 return logits
+
+            else:
+                raise ValueError("Invalid model name")
                 
 
         # if (len(sequence) > 3000 and model_name == 'gena-bert') or (len(sequence) > 10000 and model_name == 'CNN'):