Yin Fang
commited on
Commit
·
000e1c0
1
Parent(s):
a8ee13b
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,8 +60,8 @@ def sim(input_smile, output_smile):
|
|
| 60 |
|
| 61 |
|
| 62 |
def gen_process(gen_input):
|
| 63 |
-
tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen")
|
| 64 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("zjunlp/MolGen")
|
| 65 |
|
| 66 |
sf_input = tokenizer(gen_input, return_tensors="pt")
|
| 67 |
|
|
|
|
| 60 |
|
| 61 |
|
| 62 |
def gen_process(gen_input):
|
| 63 |
+
tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen-large")
|
| 64 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("zjunlp/MolGen-large")
|
| 65 |
|
| 66 |
sf_input = tokenizer(gen_input, return_tensors="pt")
|
| 67 |
|