Spaces:
Build error
Build error
| # Install Ollama first: https://ollama.ai/ | |
| import requests | |
| class LocalCoderAgent: | |
| def generate_code(self, prompt): | |
| response = requests.post( | |
| "http://localhost:11434/api/generate", | |
| json={ | |
| "model": "codellama", | |
| "prompt": f"Write Python code: {prompt}", | |
| "stream": False | |
| } | |
| ) | |
| return response.json()["response"] |