Update README.md
Browse files
README.md
CHANGED
|
@@ -76,11 +76,11 @@ model = AutoModel.from_pretrained(url, trust_remote_code=True)
|
|
| 76 |
|
| 77 |
# Input text prompts describing findings.
|
| 78 |
# The order of prompts is adjusted to capture the spectrum from absence of a finding to its temporal progression.
|
| 79 |
-
text_prompts = ["No pleural effusion or pneumothorax is seen",
|
| 80 |
-
"There is no pneumothorax or pleural effusion",
|
| 81 |
"The extent of the pleural effusion is reduced.",
|
| 82 |
"The extent of the pleural effusion remains constant.",
|
| 83 |
-
"Interval enlargement of pleural effusion"]
|
| 84 |
|
| 85 |
# Tokenize and compute the sentence embeddings
|
| 86 |
with torch.no_grad():
|
|
|
|
| 76 |
|
| 77 |
# Input text prompts describing findings.
|
| 78 |
# The order of prompts is adjusted to capture the spectrum from absence of a finding to its temporal progression.
|
| 79 |
+
text_prompts = ["No pleural effusion or pneumothorax is seen.",
|
| 80 |
+
"There is no pneumothorax or pleural effusion.",
|
| 81 |
"The extent of the pleural effusion is reduced.",
|
| 82 |
"The extent of the pleural effusion remains constant.",
|
| 83 |
+
"Interval enlargement of pleural effusion."]
|
| 84 |
|
| 85 |
# Tokenize and compute the sentence embeddings
|
| 86 |
with torch.no_grad():
|