Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ pip install transformers torch datasets pillow gradio
|
|
41 |
|
42 |
```
|
43 |
|
44 |
-
### **Load the Model
|
45 |
|
46 |
To load the model and processor for inference, use the following code:
|
47 |
|
@@ -64,7 +64,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
64 |
).to(device)
|
65 |
|
66 |
```
|
67 |
-
### **Load the
|
68 |
|
69 |
```python
|
70 |
from transformers import AutoProcessor
|
|
|
41 |
|
42 |
```
|
43 |
|
44 |
+
### **Load the Model**
|
45 |
|
46 |
To load the model and processor for inference, use the following code:
|
47 |
|
|
|
64 |
).to(device)
|
65 |
|
66 |
```
|
67 |
+
### **Load the Processor**
|
68 |
|
69 |
```python
|
70 |
from transformers import AutoProcessor
|