Update minigpt4/conversation/conversation.py

minigpt4/conversation/conversation.py

@@ -196,7 +196,7 @@ class Chat:
 
     def encode_img(self, img_list):
         image = img_list[0]
-        img_list = []
+        img_list.clear()
         if isinstance(image, str):  # is a image path
             raw_image = Image.open(image).convert('RGB')
             image = self.vis_processor(raw_image).unsqueeze(0).to(self.device)
@@ -210,14 +210,13 @@ class Chat:
 
         image_emb, _ = self.model.encode_img(image)
         img_list.append(image_emb)
-        return img_list
 
     def upload_img(self, image, conv, img_list):
         conv.append_message(conv.roles[0], "<Img><ImageHere></Img>")
         img_list.append(image)
         msg = "Received."
 
-        return msg, img_list
+        return msg
 
     def get_context_emb(self, conv, img_list):
         prompt = conv.get_prompt()