Navya-Sree commited on
Commit
2eb3524
·
verified ·
1 Parent(s): 378f180

Create Use free models

Browse files
Files changed (1) hide show
  1. Use free models +13 -0
Use free models ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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']