Spaces:
Runtime error
Runtime error
Update sample.py
Browse files
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='
|
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):
|