Update app.py

app.py

@@ -3,17 +3,11 @@ from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
 # Load the model and config when the script starts
 config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
 model = PeftModel.from_pretrained(model, "phearion/bigbrain-v0.0.1")
 
-# Move the model to the device
-model = model.to(device)
-
-
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")