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

Update ai_generator.py

Browse files
Files changed (1) hide show
  1. ai_generator.py +2 -0
ai_generator.py CHANGED
@@ -1,5 +1,7 @@
1
  from transformers import pipeline
2
 
 
 
3
  # Load the Hugging Face pipeline for text generation
4
 
5
  pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B")
 
1
  from transformers import pipeline
2
 
3
+ 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")