Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files
app.py
CHANGED
|
@@ -118,7 +118,7 @@ def load_bot(
|
|
| 118 |
|
| 119 |
|
| 120 |
bot = load_bot(
|
| 121 |
-
|
| 122 |
logging_config_path=args.logging_config_path,
|
| 123 |
model_cache_dir=args.model_cache_dir,
|
| 124 |
embedding_model_device=args.embedding_model_device,
|
|
@@ -166,12 +166,12 @@ demo = gr.ChatInterface.load(
|
|
| 166 |
container=False,
|
| 167 |
scale=7,
|
| 168 |
),
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
title="Your Personal Financial Assistant",
|
| 176 |
description="Ask me any financial or crypto market questions, and I will do my best to answer them.",
|
| 177 |
theme="soft",
|
|
|
|
| 118 |
|
| 119 |
|
| 120 |
bot = load_bot(
|
| 121 |
+
env_file_path=args.env_file_path,
|
| 122 |
logging_config_path=args.logging_config_path,
|
| 123 |
model_cache_dir=args.model_cache_dir,
|
| 124 |
embedding_model_device=args.embedding_model_device,
|
|
|
|
| 166 |
container=False,
|
| 167 |
scale=7,
|
| 168 |
),
|
| 169 |
+
additional_inputs=[
|
| 170 |
+
gr.Textbox(
|
| 171 |
+
"I am a student and I have some money that I want to invest.",
|
| 172 |
+
label="About me",
|
| 173 |
+
)
|
| 174 |
+
],
|
| 175 |
title="Your Personal Financial Assistant",
|
| 176 |
description="Ask me any financial or crypto market questions, and I will do my best to answer them.",
|
| 177 |
theme="soft",
|
bot.py
CHANGED
|
@@ -16,7 +16,7 @@ open_api_key = os.getenv("QDRANT_API_KEY")
|
|
| 16 |
|
| 17 |
|
| 18 |
def load_bot(
|
| 19 |
-
|
| 20 |
logging_config_path: str = "logging.yaml",
|
| 21 |
model_cache_dir: str = "/model_cache",
|
| 22 |
embedding_model_device: str = "cuda:0",
|
|
@@ -61,32 +61,6 @@ def load_bot(
|
|
| 61 |
return bot
|
| 62 |
|
| 63 |
|
| 64 |
-
def load_bot_dev(
|
| 65 |
-
env_file_path: str = ".env",
|
| 66 |
-
logging_config_path: str = "logging.yaml",
|
| 67 |
-
model_cache_dir: str = "./model_cache",
|
| 68 |
-
):
|
| 69 |
-
"""
|
| 70 |
-
Load the Financial Assistant Bot in dev mode: the embedding model runs on CPU and the LLM is mocked.
|
| 71 |
-
|
| 72 |
-
Args:
|
| 73 |
-
env_file_path (str): Path to the environment file.
|
| 74 |
-
logging_config_path (str): Path to the logging configuration file.
|
| 75 |
-
model_cache_dir (str): Path to the directory where the model cache is stored.
|
| 76 |
-
|
| 77 |
-
Returns:
|
| 78 |
-
The loaded Financial Assistant Bot in dev mode.
|
| 79 |
-
"""
|
| 80 |
-
|
| 81 |
-
return load_bot(
|
| 82 |
-
env_file_path=env_file_path,
|
| 83 |
-
logging_config_path=logging_config_path,
|
| 84 |
-
model_cache_dir=model_cache_dir,
|
| 85 |
-
embedding_model_device="cpu",
|
| 86 |
-
debug=True,
|
| 87 |
-
)
|
| 88 |
-
|
| 89 |
-
|
| 90 |
# === Bot Runners ===
|
| 91 |
|
| 92 |
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
def load_bot(
|
| 19 |
+
env_file_path: str = ".env",
|
| 20 |
logging_config_path: str = "logging.yaml",
|
| 21 |
model_cache_dir: str = "/model_cache",
|
| 22 |
embedding_model_device: str = "cuda:0",
|
|
|
|
| 61 |
return bot
|
| 62 |
|
| 63 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
# === Bot Runners ===
|
| 65 |
|
| 66 |
|