Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ Suitable for Arabic text generation tasks. Example usage with Transformers Pipel
|
|
41 |
```python
|
42 |
from transformers import pipeline
|
43 |
|
44 |
-
pipe = pipeline("text-generation", model="riotu-lab/ArabianGPT-
|
45 |
text = ''
|
46 |
pipe.predict(text)
|
47 |
```
|
|
|
41 |
```python
|
42 |
from transformers import pipeline
|
43 |
|
44 |
+
pipe = pipeline("text-generation", model="riotu-lab/ArabianGPT-01B", max_new_tokens=512)
|
45 |
text = ''
|
46 |
pipe.predict(text)
|
47 |
```
|