Spaces:
Build error
Build error
| # 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'] |