Update README.md
Browse files
README.md
CHANGED
@@ -63,6 +63,15 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
63 |
trust_remote_code=True
|
64 |
).to(device)
|
65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
```
|
67 |
|
68 |
### **Define the Prediction Function**
|
@@ -141,4 +150,6 @@ interface.launch()
|
|
141 |
|
142 |
Training was performed using mixed precision for efficiency.
|
143 |
|
|
|
|
|
144 |
---
|
|
|
63 |
trust_remote_code=True
|
64 |
).to(device)
|
65 |
|
66 |
+
```
|
67 |
+
### **Load the Model and Processor**
|
68 |
+
|
69 |
+
```python
|
70 |
+
from transformers import AutoProcessor
|
71 |
+
|
72 |
+
# Load the processor for the model
|
73 |
+
processor = AutoProcessor.from_pretrained("fauzail/Florence-2-VQA", trust_remote_code=True)
|
74 |
+
|
75 |
```
|
76 |
|
77 |
### **Define the Prediction Function**
|
|
|
150 |
|
151 |
Training was performed using mixed precision for efficiency.
|
152 |
|
153 |
+
|
154 |
+
|
155 |
---
|