Yashnik commited on
Commit
77963d2
1 Parent(s): 09c74ca

Update sample.py

Browse files
Files changed (1) hide show
  1. sample.py +1 -1
sample.py CHANGED
@@ -8,7 +8,7 @@ from transformers import pipeline
8
  # raise ValueError("Hugging Face token is not set in the environment variables.")
9
 
10
  # Load the model from the Hugging Face Model Hub with authentication
11
- generator = pipeline('text-generation', model='your-model-name', use_auth_token=token)
12
 
13
  # Define the prediction function
14
  def generate_text(prompt):
 
8
  # raise ValueError("Hugging Face token is not set in the environment variables.")
9
 
10
  # Load the model from the Hugging Face Model Hub with authentication
11
+ generator = pipeline('text-generation', model='bigcode/starcoder', use_auth_token=token)
12
 
13
  # Define the prediction function
14
  def generate_text(prompt):