Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
-from transformers import pipeline, GPTJForCausalLM, PeftModel, PeftConfig, AutoModelForCausalLM
-
+from transformers import pipeline, GPTJForCausalLM, AutoModelForCausalLM
+from peft import LoraConfig, get_peft_model, PeftModel, PeftConfig
 
 config = PeftConfig.from_pretrained("hackathon-somos-nlp-2023/bertin-gpt-j-6b-ner-es")
 model = AutoModelForCausalLM.from_pretrained("hackathon-somos-nlp-2023/bertin-gpt-j-6b-ner-es", return_dict=True, load_in_8bit=True, device_map='auto')