Update app.py

app.py

@@ -37,7 +37,8 @@ models = {
 
 
 # Load the mistral model
-mistral_model = MistralForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B", torch_dtype=torch.float16).to('cuda')
+mistral_model = MistralForCausalLM.from_pretrained("teknium/OpenHermes-2.5-Mistral-7B", torch_dtype=torch.float16)
+mistral_model.to('cuda')
 
 # Load ESM2 model
 model_esm, alphabet = pretrained.load_model_and_alphabet('esm2_t36_3B_UR50D')