Issue fix
Browse files
app.py
CHANGED
@@ -338,12 +338,6 @@ class SkinGPT4(nn.Module):
|
|
338 |
print(f"Aligned features : {aligned_features}")
|
339 |
print("Generated the aligned features with ViT and Qformer")
|
340 |
prompt = (
|
341 |
-
"[INST] <<SYS>>\n"
|
342 |
-
"You are a dermatology AI assistant. Analyze this skin image carefully and provide:\n"
|
343 |
-
"1. A description of visible features\n"
|
344 |
-
"2. Potential diagnoses\n"
|
345 |
-
"3. Recommendations for next steps\n"
|
346 |
-
"<</SYS>>\n\n"
|
347 |
"<Img><ImageHere></Img> Could you describe the skin disease in this image for me? [/INST]"
|
348 |
)
|
349 |
inputs = self.tokenizer(prompt, return_tensors="pt").to(images.device)
|
|
|
338 |
print(f"Aligned features : {aligned_features}")
|
339 |
print("Generated the aligned features with ViT and Qformer")
|
340 |
prompt = (
|
|
|
|
|
|
|
|
|
|
|
|
|
341 |
"<Img><ImageHere></Img> Could you describe the skin disease in this image for me? [/INST]"
|
342 |
)
|
343 |
inputs = self.tokenizer(prompt, return_tensors="pt").to(images.device)
|