Spaces:

fschwartzer
/

streamlit_chatbot

Running

fschwartzer commited on Sep 6, 2024

Commit

1f0dffd

verified ·

1 Parent(s): 40fe707

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,8 @@ import pandas as pd
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-token='HF_TOKEN'
 # Load the tokenizer
 model_name = "your-llama-model"  # Replace with the LLaMA model name

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+hf_token = os.getenv("HF_TOKEN")
+token = os.getenv("HF_TOKEN")
 # Load the tokenizer
 model_name = "your-llama-model"  # Replace with the LLaMA model name