Update app.py
Browse files
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="
|
|
|
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):
|