fix added

SkinGPT.py

@@ -232,8 +232,6 @@ class SkinGPT4(nn.Module):
     def generate(self, images, user_input=None, max_new_tokens=300):
 
         image_embeds = self.encode_image(images)
-        image_embeds_1 = self.encode_image(images)
-        print(f"Embeddings match: {torch.allclose(image_embeds, image_embeds_1, atol=1e-5)}")
 
         print(f"Aligned features : {image_embeds}")
         # print(f"\n Images embeddings shape : {image_embeds.shape} \n Llama config hidden size : {self.llama.config.hidden_size}")
@@ -249,12 +247,11 @@ class SkinGPT4(nn.Module):
             )
 
 
-        prompt = (
-            "### Instruction: <Img><IMAGE></Img> "
-            "Could you describe the skin condition in this image? "
-            "Please provide a detailed analysis including possible diagnoses. "
-            "### Response:"
-        )
+        prompt = """"### Instruction: <Img><IMAGE></Img>
+        Could you describe the skin condition in this image?
+        Please provide a detailed analysis including possible diagnoses.
+        ### Response:
+        """
 
         # prompt = """### Skin Diagnosis Analysis ###
         # <IMAGE>