Daemontatox commited on
Commit
7591bb1
·
verified ·
1 Parent(s): 1eaecfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ model_name = "Daemontatox/CogitoZ14" # Replace with your desired model
60
  tokenizer = AutoTokenizer.from_pretrained(model_name)
61
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
62
  @spaces.GPU()
63
- hf_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
64
 
65
  # LangChain LLM using Hugging Face Pipeline
66
  llm = HuggingFacePipeline(pipeline=hf_pipeline)
 
60
  tokenizer = AutoTokenizer.from_pretrained(model_name)
61
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
62
  @spaces.GPU()
63
+ hf_pipeline = pipeline("text-generation")
64
 
65
  # LangChain LLM using Hugging Face Pipeline
66
  llm = HuggingFacePipeline(pipeline=hf_pipeline)