schoginitoys commited on
Commit
deba9cd
1 Parent(s): 5bb27d2
Files changed (1) hide show
  1. app.py +2 -1
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=50, num_return_sequences=1, temperature=0.5)
 
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