Update app.py

app.py

@@ -10,11 +10,20 @@ def generateResponse(prompt):
     response = model(prompt, max_new_tokens=1024, threads=8) 
     return response
 
-with gr.Blocks() as UI:
-    gr.Markdown("This space is an attempt to run the GGUF 4 bit quantized version of 'Mistral-7B-Instruct-V0.1'")
-    prompt = gr.Textbox(label="User")
-    output = gr.Textbox(label="Assistant")
-    submitBtn = gr.Button("Submit")
-    submitBtn.click(fn=generateResponse, inputs=prompt, outputs=output, api_name="generateResponse")
+examples = ['Write a poem', 'Tell me a joke', 'Write a marketing catch phrase for an AI app']
+
+title = "Mistral-7B-Instruct-v0.1-GGUF"
+
+description = "This space is an attempt to run the GGUF 4 bit quantized version of 'Mistral-7B-Instruct-v0.1'."
+
+UI = gr.ChatInterface(
+    fn=generateResponse,
+    examples=examples,
+    title=title,
+    description=description,
+    submit_btn="Submit",
+    stop_btn="Stop generating",
+    clear_btn="Clear chat"
+)
 
 UI.queue().launch()