Fix the errors in app.py

app.py

@@ -40,3 +40,53 @@ def generate_response(user_input, chat_history):
             inputs.input_ids,
             max_length=len(inputs.input_ids[0]) + max_response_length,  # Limit the maximum length for context and response
             min_length=45,
+            temperature=0.7,  # Slightly higher temperature for more diverse responses
+            top_k=30,
+            top_p=0.9,  # Allow a bit more randomness
+            repetition_penalty=1.1,  # Mild repetition penalty
+            no_repeat_ngram_size=3,  # Ensure no repeated phrases
+            eos_token_id=tokenizer.eos_token_id,
+            pad_token_id=tokenizer.eos_token_id
+        )
+
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    # Post-process the assistant's response
+    assistant_response = response.split("Assistant:")[-1].strip()
+    # Ensure the response ends properly by stripping incomplete sentences
+    assistant_response = assistant_response.split('\n')[0].strip()
+
+    # Append the interaction to the chat history
+    chat_history.append((user_input, assistant_response))
+
+    # Return the updated chat history
+    return chat_history, chat_history
+
+def restart_chat():
+    return [], []
+
+# Create Gradio Interface
+with gr.Blocks() as chat_interface:
+    gr.Markdown("<h1><center>W.AI Chat Nikker xD</center></h1>")
+    chat_history = gr.State([])
+    with gr.Column():
+        chatbox = gr.Chatbot()
+        with gr.Row():
+            user_input = gr.Textbox(show_label=False, placeholder="Summon Wali Here...")
+            submit_button = gr.Button("Send")
+            restart_button = gr.Button("Restart")
+
+    submit_button.click(
+        generate_response,
+        inputs=[user_input, chat_history],
+        outputs=[chatbox, chat_history]
+    )
+
+    restart_button.click(
+        restart_chat,
+        inputs=[],
+        outputs=[chatbox, chat_history]
+    )
+
+# Launch the Gradio interface with share=True
+chat_interface.launch(share=True)