Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModel, AutoModelForCausalLM, pipeline
+from sentence_transformers import SentenceTransformer
 import time
 from datetime import datetime
 import json