Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ class DocumentGenerator:
|
|
96 |
"""Gerencia a geração de documentos usando GPT4All-J"""
|
97 |
|
98 |
def __init__(self):
|
99 |
-
self.pipe = pipeline("text-generation", model="
|
100 |
self.prompt_manager = PromptManager()
|
101 |
|
102 |
def generate(self, doc_type: str, context: Dict[str, str]) -> str:
|
|
|
96 |
"""Gerencia a geração de documentos usando GPT4All-J"""
|
97 |
|
98 |
def __init__(self):
|
99 |
+
self.pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
|
100 |
self.prompt_manager = PromptManager()
|
101 |
|
102 |
def generate(self, doc_type: str, context: Dict[str, str]) -> str:
|