ANASAKHTAR commited on
Commit
4dd4685
·
verified ·
1 Parent(s): f8c10ae

Update ai_generator.py

Browse files
Files changed (1) hide show
  1. ai_generator.py +1 -1
ai_generator.py CHANGED
@@ -4,7 +4,7 @@ huggingface-cli login
4
 
5
  # Load the Hugging Face pipeline for text generation
6
 
7
- pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B")
8
 
9
 
10
 
 
4
 
5
  # Load the Hugging Face pipeline for text generation
6
 
7
+ pipe = pipeline("text-generation", model="nvidia/Llama-3.1-Nemotron-70B-Instruct-HF")
8
 
9
 
10