Update app.py

app.py

@@ -13,15 +13,20 @@ pipe = pipeline("text-generation", model=model_name, device="cuda")
 generate_kwargs = {'max_new_tokens': 20}
 
 
-system_prompt = '''You are given a partial input text for a chat interface. Propose auto-completion to the text. You have several roles:
+system_prompt = '''You are given a partial input text for another AI chat interface. 
+Propose auto-completion to the text. You have several roles:
 - Fight under-specification.
 - Complete text to save the user time.
 
 Don't suggest anything if there are no good suggestions. 
-Make sure the suggestions are valid completions of the text! No need for them to complete the text completely. 
-Suggest only up to 5 works ahead. The scheme of your answer should be "answer1;answer2;answer3" (return between 0 to 4 answers).
+Make sure the suggestions are valid completions of the text! Suggest only up to 5 words ahead. The scheme of your answer should be "answer1;answer2;answer3" (return between 0 to 4 answers).
 Answers should be only the completions themselves. 
+You will now get a blank message from the user and then after your answer, the user will give you the text to complete.
+
+'''
 
+
+extra_prompt = '''
 Examples: 
 (1)
 User: "Help me write a sentiment analysis pipeline"
@@ -34,7 +39,6 @@ Assistant: "" (nothing much to contribute at this point. return nothing)
 (3)
 User: "Help me find a present for my"
 Assistant: "girlfriend;mother;father;friend"
-You will now get a blank message from the user and then after your answer, the user will give you the text to complete.
 '''