Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
+import
+ torch
 
 # Device configuration (prioritize GPU if available)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load models and tokenizer efficiently
-model_id = "phearion/bigbrain-v0.0.1"
-config = PeftConfig.from_pretrained(model_id=model_id)
+config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 model = PeftModel.from_pretrained(model_id)
 model.to(device)