Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
3 |
import torch
|
4 |
|
5 |
# Hugging Face repository details
|
6 |
-
MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
|
|
|
7 |
|
8 |
def load_model():
|
9 |
"""Load the Hugging Face model and tokenizer."""
|
|
|
3 |
import torch
|
4 |
|
5 |
# Hugging Face repository details
|
6 |
+
# MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
|
7 |
+
MODEL_ID = "meta-llama/Llama-3.1-8B-Instruct"
|
8 |
|
9 |
def load_model():
|
10 |
"""Load the Hugging Face model and tokenizer."""
|