DHEIVER commited on
Commit
5463acc
·
verified ·
1 Parent(s): e411978

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="nomic-ai/gpt4all-j")
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: