Spaces:
Runtime error
Runtime error
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):
|