Spaces:

junaidbaber
/

demo_lowcode_llm

Sleeping

junaidbaber commited on Jan 15

Commit

df369f2

verified ·

1 Parent(s): a7a70df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 # Hugging Face repository details
-MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
 def load_model():
     """Load the Hugging Face model and tokenizer."""

 import torch
 # Hugging Face repository details
+# MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
+MODEL_ID = "meta-llama/Llama-3.1-8B-Instruct"
 def load_model():
     """Load the Hugging Face model and tokenizer."""