Update app.py

app.py

@@ -9,7 +9,7 @@ model = GPTJForCausalLM.from_pretrained("hackathon-somos-nlp-2023/bertin-gpt-j-6
 tokenizer = AutoTokenizer.from_pretrained("hackathon-somos-nlp-2023/bertin-gpt-j-6b-ner-es")
 
 # create pipeline
-pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, adapter_names=["adapter"], adapter_setup=adapter_setup, device=0,)
 
 def predict(text):
   return pipe(f"text: {text}, entities:")["generated_text"]