Rafa1986 commited on
Commit
bb7abe3
·
verified ·
1 Parent(s): 0dd6963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import pipeline
14
  HF_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
15
 
16
  # Carregar o modelo Mistral 7B gratuitamente do Hugging Face
17
- chatbot_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1", token=HF_API_TOKEN)
18
 
19
  def extract_files_from_folder(folder_path):
20
  """Scans a folder for PDF, TXT, CSV, DOCX, and IPYNB files."""
 
14
  HF_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
15
 
16
  # Carregar o modelo Mistral 7B gratuitamente do Hugging Face
17
+ chatbot_pipeline = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", token=HF_API_TOKEN)
18
 
19
  def extract_files_from_folder(folder_path):
20
  """Scans a folder for PDF, TXT, CSV, DOCX, and IPYNB files."""