AMfeta99 commited on
Commit
11cd28e
·
verified ·
1 Parent(s): 651b5d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -60,7 +60,8 @@ text_to_image_tool = TextToImageTool(client=text_to_image_client)
60
  search_tool = DuckDuckGoSearchResults()
61
 
62
  # Load a public, token-free model locally via transformers pipeline
63
- text_gen_pipeline = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_new_tokens=512)
 
64
 
65
  # Wrap pipeline into a LangChain LLM
66
  class PipelineLLM(LLM):
 
60
  search_tool = DuckDuckGoSearchResults()
61
 
62
  # Load a public, token-free model locally via transformers pipeline
63
+ text_gen_pipeline = pipeline("text-generation", model="Qwen/Qwen2.5-72B-Instruct", max_new_tokens=512)
64
+ #tiiuae/falcon-7b-instruct
65
 
66
  # Wrap pipeline into a LangChain LLM
67
  class PipelineLLM(LLM):