Update app.py

app.py

@@ -183,7 +183,7 @@ def analyze_dna(username, password, sequence, model_name):
         #             logits_sum += logits
         #     logits_avg = logits_sum / num_shifts
         # else:
-        logits_avg = get_logits(sequence)
+        logits_avg = get_logits(sequence, model_name)
 
         probabilities = torch.nn.functional.softmax(logits_avg, dim=-1).squeeze().tolist()
         top_5_indices = sorted(range(len(probabilities)), key=lambda i: probabilities[i], reverse=True)[:5]