Update README.md
Browse files
README.md
CHANGED
|
@@ -181,6 +181,7 @@ for i, prob in enumerate(probs[0]):
|
|
| 181 |
```
|
| 182 |
|
| 183 |
**B. Loading Disease Detection Models**
|
|
|
|
| 184 |
Note: The disease model takes symptom feature vectors as input (Shape: [Batch, Seq_Len, 38]), not raw text.
|
| 185 |
```python
|
| 186 |
import torch
|
|
|
|
| 181 |
```
|
| 182 |
|
| 183 |
**B. Loading Disease Detection Models**
|
| 184 |
+
|
| 185 |
Note: The disease model takes symptom feature vectors as input (Shape: [Batch, Seq_Len, 38]), not raw text.
|
| 186 |
```python
|
| 187 |
import torch
|