Commit
•
deba9cd
1
Parent(s):
5bb27d2
Sree
Browse files
app.py
CHANGED
@@ -6,7 +6,8 @@ generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
|
|
6 |
|
7 |
def generate_code(prompt):
|
8 |
# Generate code from the model
|
9 |
-
responses = generator(prompt, max_length=
|
|
|
10 |
return responses[0]['generated_text']
|
11 |
|
12 |
|
|
|
6 |
|
7 |
def generate_code(prompt):
|
8 |
# Generate code from the model
|
9 |
+
responses = generator(prompt, max_length=500, num_return_sequences=1, temperature=0.1)
|
10 |
+
# responses = generator(prompt, max_length=50, num_return_sequences=1, temperature=0.5)
|
11 |
return responses[0]['generated_text']
|
12 |
|
13 |
|