frimponge commited on
Commit
b598360
·
verified ·
1 Parent(s): dc2bbc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ class PoultryFarmBot:
106
  bot = PoultryFarmBot()
107
 
108
  # Load Llama 2 model and tokenizer for text generation
109
- model_name = "meta-llama/Llama-2-7b-chat-hf" # Example model name; replace with the correct Llama 2 model path
110
  tokenizer = AutoTokenizer.from_pretrained(model_name)
111
  model = AutoModelForCausalLM.from_pretrained(model_name)
112
 
 
106
  bot = PoultryFarmBot()
107
 
108
  # Load Llama 2 model and tokenizer for text generation
109
+ model_name = "meta-llama/Llama-2-7b-chat-hf"
110
  tokenizer = AutoTokenizer.from_pretrained(model_name)
111
  model = AutoModelForCausalLM.from_pretrained(model_name)
112