Upload folder using huggingface_hub
Browse files- LLM/language_model.py +2 -0
LLM/language_model.py
CHANGED
|
@@ -111,6 +111,8 @@ class LanguageModelHandler(BaseHandler):
|
|
| 111 |
)
|
| 112 |
|
| 113 |
def process(self, prompt):
|
|
|
|
|
|
|
| 114 |
logger.debug("infering language model...")
|
| 115 |
language_code = None
|
| 116 |
if isinstance(prompt, tuple):
|
|
|
|
| 111 |
)
|
| 112 |
|
| 113 |
def process(self, prompt):
|
| 114 |
+
console.print("infering language model...")
|
| 115 |
+
console.print("prompt")
|
| 116 |
logger.debug("infering language model...")
|
| 117 |
language_code = None
|
| 118 |
if isinstance(prompt, tuple):
|