Update app.py

app.py

@@ -1,9 +1,5 @@
-from transformers import BertConfig, BertModel
+from transformers import AutoModel
 
-# Building the config
-config = BertConfig()
+checkpoint = "meta-llama/Llama-2-7b"
+model = AutoModel.from_pretrained(checkpoint)
 
-# Building the model from the config
-model = BertModel(config)
-
-print(config)