Navya-Sree commited on
Commit
eb300c2
·
verified ·
1 Parent(s): 060bfc1

Delete Use free models

Browse files
Files changed (1) hide show
  1. Use free models +0 -13
Use free models DELETED
@@ -1,13 +0,0 @@
1
- # Alternative: Use free models
2
- from transformers import pipeline
3
-
4
- class FreeCoderAgent:
5
- def __init__(self):
6
- self.generator = pipeline(
7
- "text-generation",
8
- model="codellama/CodeLlama-7b-hf"
9
- )
10
-
11
- def generate_code(self, prompt):
12
- result = self.generator(f"Write Python code: {prompt}", max_length=200)
13
- return result[0]['generated_text']