fix added

SkinGPT.py

@@ -249,18 +249,18 @@ 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>
-        Could you describe the skin condition in this image?
-        Please provide a detailed analysis including possible diagnoses.
-        ### Response:"""
+        # prompt = """### Skin Diagnosis Analysis ###
+        # <IMAGE>
+        # Could you describe the skin condition in this image?
+        # Please provide a detailed analysis including possible diagnoses.
+        # ### Response:"""
 
 
         # print(f"\n[DEBUG] Raw Prompt:\n{prompt}")
@@ -271,8 +271,11 @@ class SkinGPT4(nn.Module):
             padding_side="right"
         )
         # self.tokenizer.add_special_tokens({'additional_special_tokens': ['<Img>', '</Img>', '<ImageHere>']})
+        # num_added = self.tokenizer.add_special_tokens({
+        #     'additional_special_tokens': ['<IMAGE>']
+        # })
         num_added = self.tokenizer.add_special_tokens({
-            'additional_special_tokens': ['<IMAGE>']
+            'additional_special_tokens': ['<Img>', '</Img>', '<IMAGE>']
         })
 
         if num_added == 0: