Decrease temperature

app.py

@@ -126,7 +126,7 @@ def predict_genus(method: str, dna_sequence: str, latitude: str, longitude: str)
     if method == "fine_tuned_model":
         bert_inputs = tokenize(dna_sequence)
         logits = classification_model(bert_inputs, torch.zeros(1, 7))
-        temperature = 0.5
+        temperature = 0.2
         probs = torch.softmax(logits / temperature, dim=1).squeeze()
         top_k = torch.topk(probs, 10)
         top_k = pd.Series(