Commit
•
66e0337
1
Parent(s):
51472ac
Update README.md
Browse files
README.md
CHANGED
@@ -35,8 +35,8 @@ import torch
|
|
35 |
from transformers import NougatProcessor, VisionEncoderDecoderModel
|
36 |
|
37 |
# Load the model and processor
|
38 |
-
processor = NougatProcessor.from_pretrained("
|
39 |
-
model = VisionEncoderDecoderModel.from_pretrained("arabic-small-
|
40 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
41 |
model.to(device)
|
42 |
|
|
|
35 |
from transformers import NougatProcessor, VisionEncoderDecoderModel
|
36 |
|
37 |
# Load the model and processor
|
38 |
+
processor = NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat")
|
39 |
+
model = VisionEncoderDecoderModel.from_pretrained("MohamedRashad/arabic-small-nougat")
|
40 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
41 |
model.to(device)
|
42 |
|