Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def greet1(name):
|
|
17 |
attention_mask=sf_input["attention_mask"],
|
18 |
max_length=15,
|
19 |
min_length=5,
|
20 |
-
num_return_sequences=
|
21 |
num_beams=5)
|
22 |
|
23 |
sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
|
@@ -34,7 +34,7 @@ def greet2(name):
|
|
34 |
attention_mask=sf_input["attention_mask"],
|
35 |
max_length=15,
|
36 |
min_length=5,
|
37 |
-
num_return_sequences=
|
38 |
num_beams=5)
|
39 |
|
40 |
sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
|
|
|
17 |
attention_mask=sf_input["attention_mask"],
|
18 |
max_length=15,
|
19 |
min_length=5,
|
20 |
+
num_return_sequences=4,
|
21 |
num_beams=5)
|
22 |
|
23 |
sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
|
|
|
34 |
attention_mask=sf_input["attention_mask"],
|
35 |
max_length=15,
|
36 |
min_length=5,
|
37 |
+
num_return_sequences=4,
|
38 |
num_beams=5)
|
39 |
|
40 |
sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
|