multi-agent-code-assistant / Use free models
Navya-Sree's picture
Create Use free models
2eb3524 verified
raw
history blame
399 Bytes
# Alternative: Use free models
from transformers import pipeline
class FreeCoderAgent:
def __init__(self):
self.generator = pipeline(
"text-generation",
model="codellama/CodeLlama-7b-hf"
)
def generate_code(self, prompt):
result = self.generator(f"Write Python code: {prompt}", max_length=200)
return result[0]['generated_text']