Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -30,10 +30,17 @@ End of the texts
|
|
30 |
{target} translated texts:
|
31 |
|
32 |
"""
|
33 |
-
|
34 |
model_kwargs={
|
35 |
"max_new_tokens":2048,
|
36 |
"temperature": 0.01,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
"truncate": 1000,
|
38 |
"seed" : 256,
|
39 |
"stop" : ["</s>","<|endoftext|>","<|end|>"],
|
@@ -41,7 +48,7 @@ model_kwargs={
|
|
41 |
|
42 |
llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
|
43 |
llm2 = HuggingFaceHub(repo_id=starchat_repo, task="text-generation", model_kwargs=model_kwargs)
|
44 |
-
llm3 = HuggingFaceHub(repo_id=bloom_repo, task="text-generation", model_kwargs=
|
45 |
|
46 |
def translation(source, target, text):
|
47 |
response = text
|
|
|
30 |
{target} translated texts:
|
31 |
|
32 |
"""
|
|
|
33 |
model_kwargs={
|
34 |
"max_new_tokens":2048,
|
35 |
"temperature": 0.01,
|
36 |
+
"truncate": 4096,
|
37 |
+
"seed" : 256,
|
38 |
+
"stop" : ["</s>","<|endoftext|>","<|end|>"],
|
39 |
+
}
|
40 |
+
|
41 |
+
bloom_model_kwargs={
|
42 |
+
"max_new_tokens":1000,
|
43 |
+
"temperature": 0.01,
|
44 |
"truncate": 1000,
|
45 |
"seed" : 256,
|
46 |
"stop" : ["</s>","<|endoftext|>","<|end|>"],
|
|
|
48 |
|
49 |
llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
|
50 |
llm2 = HuggingFaceHub(repo_id=starchat_repo, task="text-generation", model_kwargs=model_kwargs)
|
51 |
+
llm3 = HuggingFaceHub(repo_id=bloom_repo, task="text-generation", model_kwargs=bloom_model_kwargs)
|
52 |
|
53 |
def translation(source, target, text):
|
54 |
response = text
|