switched model
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
| 3 |
|
| 4 |
# Initialize the StarCoder model
|
| 5 |
# Replace 'models/bigcode/starcoder' with the actual model path if different
|
| 6 |
-
code_generator = pipeline('text-generation', model='
|
| 7 |
|
| 8 |
def generate_code(prompt):
|
| 9 |
#"""
|
|
|
|
| 3 |
|
| 4 |
# Initialize the StarCoder model
|
| 5 |
# Replace 'models/bigcode/starcoder' with the actual model path if different
|
| 6 |
+
code_generator = pipeline('text-generation', model='codellama/CodeLlama-7b-hf')
|
| 7 |
|
| 8 |
def generate_code(prompt):
|
| 9 |
#"""
|