Update app.py

app.py

@@ -35,13 +35,13 @@ def respond(encoded_smiles, max_tokens, temperature, top_p, top_k):
         }
 
         # Tokenize the input
-        input_ids = llm.tokenizer(encoded_smiles)
+        input_ids = llm.tokenize(encoded_smiles)
 
         # Generate the output
         outputs = llm.generate(input_ids=input_ids, **settings)
 
         # Decode the output tokens to text
-        output_text = llm.tokenizer.decode(outputs[0])
+        output_text = llm.decode(outputs[0], skip_special_tokens=True)
 
         # Extract the predicted selfies from the output text
         first_inst_index = output_text.find("[/INST]")