Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from transformers import (AutoModelForCausalLM, AutoTokenizer,
+from transformers import (AutoModel, AutoModelForCausalLM, AutoTokenizer,
                           LlamaForCausalLM, LlamaTokenizer)
 
 title = "🤖AI HeavyMetal-ChatBot"
@@ -10,8 +10,12 @@ examples = [["How are you?"]]
 
 # tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 # model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
-tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/llama-tokenizer")
-model = LlamaForCausalLM.from_pretrained("hf-internal-testing/llama-tokenizer")
+# tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/llama-tokenizer")
+# model = LlamaForCausalLM.from_pretrained("hf-internal-testing/llama-tokenizer")
+
+model = "meta-llama/Llama-2-7b-chat-hf"
+
+tokenizer = AutoTokenizer.from_pretrained(model)
 
 
 def predict(input, history=[]):