Update inference_server.py

inference_server.py

@@ -28,8 +28,7 @@ def predict(input_text: str):
         logits = output[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.csine_similarity(logits[:, 0, :],
                                             logits[:, -1, :], dim=1)
     _, prediction_index = torch.max(similarity_scores, dim=0)
     prediction = list(strings)[prediction_index]