Update app.py

app.py

@@ -21,6 +21,9 @@ def QLoRA_Chatgpt(prompt):
     result = pipe(f"<s>[INST] {prompt} [/INST]")
     return(result[0]['generated_text'])
     # return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=QLoRA_Chatgpt, inputs=gr.Textbox("how can help you today", label='prompt'), outputs=gr.Textbox(label='Generated-output'))
+# Define Interface
+description = 'An AI assistant that works on the Microsoft Phi 2 model, which has been trained specifically on the Open Assistant dataset using the QLora method, operates effectively.Link to the model: https://huggingface.co/microsoft/phi-2 Link to the dataset: https://huggingface.co/datasets/OpenAssistant/oasst1  '
+title = 'AI Chat bot finetuned on Microsoft Phi 2 model using QLORA'
+iface = gr.Interface(fn=QLoRA_Chatgpt, inputs=gr.Textbox("how can help you today", label='prompt'), outputs=gr.Textbox(label='Generated-output',scale = 2), title = title,
+                    description = description)
 iface.launch(share=True)