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 |
|