Update main.py

main.py

@@ -1,8 +1,9 @@
-import torch
 from flask import Flask, request, jsonify
 from langchain_community.llms import LlamaCpp
 from sentence_transformers import SentenceTransformer
 from transformers import AutoModel
+import torch
+from torch.nn.functional import cosine_similarity
 import os 
 
 app = Flask(__name__)