Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-from peft import PeftModel
-, PeftConfig
+from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
@@ -8,8 +7,7 @@ import torch
 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")
+model = PeftModel.from_pretrained(model,"phearion/bigbrain-v0.0.1")
 
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")