Update tools/arabic_generator.py
Browse files
tools/arabic_generator.py
CHANGED
|
@@ -5,7 +5,6 @@ class ArabicTextGenerator:
|
|
| 5 |
self.model_loaded = False
|
| 6 |
|
| 7 |
def load_model(self):
|
| 8 |
-
"""Lazy load the model to improve startup time"""
|
| 9 |
if not self.model_loaded:
|
| 10 |
self.generator = pipeline(
|
| 11 |
'text-generation',
|
|
|
|
| 5 |
self.model_loaded = False
|
| 6 |
|
| 7 |
def load_model(self):
|
|
|
|
| 8 |
if not self.model_loaded:
|
| 9 |
self.generator = pipeline(
|
| 10 |
'text-generation',
|