Update app.py
Browse files
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"
|
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 |
|