Spaces:
Runtime error
Runtime error
| from huggingface_hub import HfFileSystem | |
| fs = HfFileSystem() | |
| from langchain import HuggingFaceHub | |
| from langchain import SQLDatabase | |
| from langchain_experimental.sql.base import SQLDatabaseSequentialChain | |
| from langchain_experimental.sql import SQLDatabaseChain | |
| import json | |
| import os | |
| overal_temperature = 0.2 | |
| os.environ["HUGGINGFACEHUB_API_TOKEN"] = "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" | |
| llm = HuggingFaceHub(repo_id="juierror/flan-t5-text2sql-with-schema-v2", | |
| model_kwargs={"temperature":overal_temperature, | |
| "max_new_tokens":200} | |
| ) | |
| sqlite_db_path = "hf://datasets/ankit75100/Ankit/test_db.db" | |
| db1 = SQLDatabase.from_uri(f"sqlite:///{sqlite_db_path}") | |
| include_tables=['Employee','Album','Artist','Customer'], | |
| db_chain = SQLDatabaseChain.from_llm(llm, db1, verbose=True, return_direct=True) | |
| #user_input = input("Enter something: ") | |
| #datbase=db_chain.run(user_input) | |
| #print(datbase) | |
| def process_input(input_text): | |
| Result=db_chain.run(input_text) | |
| return Result | |