Update app.py

app.py

@@ -59,6 +59,7 @@ retriever = db.as_retriever(
 model_name = "Daemontatox/CogitoZ14"  # Replace with your desired model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
+@spaces.GPU()
 hf_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 # LangChain LLM using Hugging Face Pipeline