Update inference_server.py

inference_server.py

@@ -26,7 +26,7 @@ def predict(input_text: str):
         logits = outputs[0]
 
     # Find the input string that is most similar to the new input string, according to the BERT LM
-    similarity_scores = torch.nn.functional.csine_similarity(logits[:, 0, :],
+    similarity_scores = torch.nn.functional.cosine_similarity(logits[:, 0, :],
                                             logits[:, -1, :], dim=1)
     _, prediction_index = torch.max(similarity_scores, dim=0)
     prediction = list(strings)[prediction_index]