Update app.py

app.py

@@ -12,13 +12,16 @@ from huggingface_hub import hf_hub_download
 
 # Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
 model = AutoModelForCausalLM.from_pretrained("TheBloke/Mistral-7B-Instruct-v0.1-GGUF", model_file="mistral-7b-instruct-v0.1.Q5_K_S.gguf", model_type="mistral", gpu_layers=0)
-ins = '''[INST] <<SYS>>
+"""ins = '''[INST] <<SYS>>
 You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe.  Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. Please ensure that your responses are socially unbiased and positive in nature.
 If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information.
 <</SYS>>
 {} [/INST]
 '''
-
+"""
+ins=""" System:Your Helpful Assistant
+        {User}:{question}
+        {bot_name}:"""
 theme = gr.themes.Monochrome(
     primary_hue="indigo",
     secondary_hue="blue",
@@ -26,8 +29,8 @@ theme = gr.themes.Monochrome(
     radius_size=gr.themes.sizes.radius_sm,
     font=[gr.themes.GoogleFont("Open Sans"), "ui-sans-serif", "system-ui", "sans-serif"],
 )
-def response(question):
-    res = model(ins.format(question))
+def response(user_name,bot_name,question):
+    res = model(ins.format(user_name=user_name,bot_name,bot_namequestion=question))
     yield res
 
 
@@ -107,6 +110,8 @@ with gr.Blocks(theme=seafoam, analytics_enabled=False, css=css) as demo:
         )
 
         with gr.Row():
+            user_name = gr.Textbox(placeholder="Enter your Name", label="User Name", elem_id="user-input")
+            bot_name = gr.Textbox(placeholder="Enter your question here", label="Question", elem_id="q-input")
             with gr.Column(scale=3):
                 instruction = gr.Textbox(placeholder="Enter your question here", label="Question", elem_id="q-input")
 
@@ -116,7 +121,7 @@ with gr.Blocks(theme=seafoam, analytics_enabled=False, css=css) as demo:
                 submit = gr.Button("Generate", variant="primary")
                 gr.Examples(
                     examples=examples,
-                    inputs=[instruction],
+                    inputs=[user_name,bot_name,instruction],
                     cache_examples=True,
                     fn=process_example,
                     outputs=[output],
@@ -124,7 +129,7 @@ with gr.Blocks(theme=seafoam, analytics_enabled=False, css=css) as demo:
         
 
 
-    submit.click(response, inputs=[instruction], outputs=[output])
-    instruction.submit(response, inputs=[instruction], outputs=[output])
+    submit.click(response, inputs=[use_name,bot_name,instruction], outputs=[output])
+    instruction.submit(response, inputs=[use_name,bot_name,instruction], outputs=[output])
 
 demo.queue(concurrency_count=1).launch(debug=False)