Spaces:

JerniganLab
/

interview-and-qa

Sleeping

tolulope commited on Jan 4

Commit

30ae265

verified ·

1 Parent(s): ead7acd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 model_id = "JerniganLab/interviews-and-qa"
 base_model = "meta-llama/Meta-Llama-3-8B-Instruct"
-llama_model = transformers.AutoModelForCausalLM(base_model)
 pipeline = transformers.pipeline(
     "text-generation",

 model_id = "JerniganLab/interviews-and-qa"
 base_model = "meta-llama/Meta-Llama-3-8B-Instruct"
+llama_model = transformers.AutoModelForCausalLM.from_pretrained(base_model)
 pipeline = transformers.pipeline(
     "text-generation",