Update chatbot.py

chatbot.py

@@ -40,12 +40,6 @@ class Chatbot:
 
     def load_models(self):
         self.model = SentenceTransformer('clip-ViT-B-32')
-        self.bert_model_name = "bert-base-uncased"
-        self.bert_model = BertModel.from_pretrained(self.bert_model_name)
-        self.bert_tokenizer = BertTokenizer.from_pretrained(self.bert_model_name)
-        self.gpt2_model_name = "gpt2"
-        self.gpt2_model = GPT2LMHeadModel.from_pretrained(self.gpt2_model_name)
-        self.gpt2_tokenizer = GPT2Tokenizer.from_pretrained(self.gpt2_model_name)
         self.blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
         self.blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
 
@@ -81,7 +75,7 @@ class Chatbot:
             docs.append(doc)
         return docs
 
-    def get_results(self, query, embeddings, top_k=10):
+    def get_results(self, query, embeddings, top_k=5):
         query_embedding = self.model.encode([query])
         cos_scores = util.pytorch_cos_sim(query_embedding, embeddings)[0]
         top_results = torch.topk(cos_scores, k=top_k)