Arko007 commited on
Commit
3c7995c
·
verified ·
1 Parent(s): 05fb6b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,7 +61,7 @@ def generate_code(instruction):
61
 
62
  outputs = model.generate(
63
  **inputs,
64
- max_new_tokens=1024, # Give it plenty of room to write
65
  temperature=0.1,
66
  top_p=0.9,
67
  eos_token_id=tokenizer.eos_token_id
@@ -88,4 +88,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
88
  generate_button.click(fn=generate_code, inputs=instruction_box, outputs=output_box)
89
 
90
  # This will launch the app when deployed on Hugging Face Spaces
91
- demo.launch()
 
61
 
62
  outputs = model.generate(
63
  **inputs,
64
+ max_new_tokens=2048, # Increased token limit for longer code generation
65
  temperature=0.1,
66
  top_p=0.9,
67
  eos_token_id=tokenizer.eos_token_id
 
88
  generate_button.click(fn=generate_code, inputs=instruction_box, outputs=output_box)
89
 
90
  # This will launch the app when deployed on Hugging Face Spaces
91
+ demo.launch(share=True)