Update app.py

app.py

@@ -1,6 +1,11 @@
 import streamlit as st
 from transformers import AutoModel, AutoTokenizer
+from transformers import LlamaTokenizer, LlamaForCausalLM
+from biomistral import BioMistral
 
+tokenizer = LlamaTokenizer.from_pretrained("BioMistral/BioMistral-7B")
+model = LlamaForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
+biomistral = BioMistral(model=model, tokenizer=tokenizer)
 # Load BioMistral model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("BioMistral/BioMistral-7B")
 model = AutoModel.from_pretrained("BioMistral/BioMistral-7B")