Extending classification

app.py

@@ -239,6 +239,8 @@ def text_to_text_generation(verbose, prompt):
         f"response (0/1/2): 2 \n"
         f"text: What is a LoadBalancer in Kubernetes?\n"
         f"response (0/1/2): 1 \n"
+        f"text: How can I enhance the performance of a k8s cluster?\n"
+        f"response (0/1/2): 1 \n"
         f'Classify the following: [/INST] \ntext: "{prompt}\n"'
         f"response (0/1/2): "
     )