sadgaj commited on
Commit
df2422f
·
1 Parent(s): a03b1ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def greet(name):
64
  tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen-large-opt")
65
  model = AutoModelForSeq2SeqLM.from_pretrained("zjunlp/MolGen-large-opt")
66
 
67
- input = "[C][C][=Branch1][C][=O][N][C][C][O][C][C][O][C][C][O][C][C][Ring1][N]"
68
 
69
  sf_input = tokenizer(input, return_tensors="pt")
70
  molecules = model.generate(
 
64
  tokenizer = AutoTokenizer.from_pretrained("zjunlp/MolGen-large-opt")
65
  model = AutoModelForSeq2SeqLM.from_pretrained("zjunlp/MolGen-large-opt")
66
 
67
+ input = name
68
 
69
  sf_input = tokenizer(input, return_tensors="pt")
70
  molecules = model.generate(