Update app.py

app.py

@@ -13,6 +13,7 @@ End of the document.[/INST]
 {target} translated document:
 
 """
+
 starchat_template = """<|system|>I want you to act as document language translator. You do translation {source} texts in document into then you return to me the translated document AND DO NOTHING ELSE.<</SYS>>
 Begin of the document:
 {query}
@@ -22,12 +23,10 @@ End of the document<|end|>
 
 """
 
-starchat_template = """Translation {source} texts into {target}. then you return to me the translated document AND DO NOTHING ELSE.<</SYS>>
-Begin of the texts:
+bloom_template = """Translate bellow {source} text into {target}:
 {query}
-End of the texts
-
-{target} translated texts:
+======
+{target} translated text:
 
 """
 model_kwargs={
@@ -43,7 +42,7 @@ bloom_model_kwargs={
             "temperature": 0.01,
             "truncate": 1000,
             "seed" : 256,
-            "stop" : ["</s>","<|endoftext|>","<|end|>"],
+            "stop" : ["</s>","<|endoftext|>","<|end|>","======"],
             }
 
 llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
@@ -53,7 +52,7 @@ llm3 = HuggingFaceHub(repo_id=bloom_repo, task="text-generation", model_kwargs=b
 def translation(source, target, text):
     response = text
     try:
-        input_prompt = llamma_template.replace("{source}", source)
+        input_prompt = bloom_template.replace("{source}", source)
         input_prompt = input_prompt.replace("{target}", target)
         input_prompt = input_prompt.replace("{query}", text)
         response=llm3(input_prompt)