Update app.py

app.py

@@ -24,10 +24,10 @@ def generate_documentation(code_input):
     model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
     
     # Generate the documentation
-    generated_ids = model.generate(**model_inputs, max_new_tokens=512)
+    generated_ids = model.generate(**model_inputs, max_new_tokens=4000)
     generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)]
     documentation = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    
+    print(documentation)
     return documentation
 
 # Function to generate and download PDF
@@ -52,7 +52,7 @@ def process_code(code_input):
 # Set up the Gradio app with Bootstrap, icons, and smiley
 with gr.Blocks(css=".container { font-family: 'Roboto', sans-serif; } .btn-primary { background-color: #007bff; } .icon { margin-right: 10px; }") as app:
     gr.Markdown("""
-    # :notebook_with_decorative_cover: Code Documentation Generator
+    #Code Documentation Generator
     
     Paste your code below, and the app will generate the README and detailed documentation for you.
     The output will also be available for download as a PDF.
@@ -62,7 +62,7 @@ with gr.Blocks(css=".container { font-family: 'Roboto', sans-serif; } .btn-prima
         code_input = gr.Textbox(lines=10, label="Paste your code here", placeholder="Enter your code...", show_label=False, elem_classes="form-control")
     
     with gr.Row():
-        generate_button = gr.Button(":sparkles: Generate Documentation", elem_classes="btn btn-primary")
+        generate_button = gr.Button("Generate Documentation", elem_classes="btn btn-primary")
     
     with gr.Row():
         output_text = gr.Textbox(label="Generated Documentation", lines=20, interactive=False)