Update data.py
Browse files
data.py
CHANGED
|
@@ -15,7 +15,7 @@ import time
|
|
| 15 |
# accessing and defining the llm served in the cluster
|
| 16 |
from langchain_community.chat_models import ChatOpenAI
|
| 17 |
|
| 18 |
-
NGROK_URL = "https://
|
| 19 |
VLLM_MODEL_NAME = "hugging-quants/Meta-Llama-3.1-8B-Instruct-AWQ-INT4"
|
| 20 |
|
| 21 |
llm1 = ChatOpenAI(
|
|
@@ -179,7 +179,7 @@ def debug_print(category, message):
|
|
| 179 |
print(f"[{timestamp}] [{category}] {message}", flush=True, file=sys.stdout)
|
| 180 |
|
| 181 |
|
| 182 |
-
#
|
| 183 |
eligibility_df = pd.read_csv("cards_eligibility_updated.csv")
|
| 184 |
df = pd.read_csv("credit_card_data_updated.csv")
|
| 185 |
# print(df.head())
|
|
@@ -190,7 +190,7 @@ all_card_names = df_all_cards["name"].tolist()
|
|
| 190 |
all_card_lookup = dict(zip(df_all_cards["name"], df_all_cards["description"]))
|
| 191 |
|
| 192 |
|
| 193 |
-
#
|
| 194 |
eligibility_lookup = {}
|
| 195 |
for _, row in eligibility_df.iterrows():
|
| 196 |
name = row["Name"].strip()
|
|
|
|
| 15 |
# accessing and defining the llm served in the cluster
|
| 16 |
from langchain_community.chat_models import ChatOpenAI
|
| 17 |
|
| 18 |
+
NGROK_URL = "https://c89a68ddd25a.ngrok-free.app"
|
| 19 |
VLLM_MODEL_NAME = "hugging-quants/Meta-Llama-3.1-8B-Instruct-AWQ-INT4"
|
| 20 |
|
| 21 |
llm1 = ChatOpenAI(
|
|
|
|
| 179 |
print(f"[{timestamp}] [{category}] {message}", flush=True, file=sys.stdout)
|
| 180 |
|
| 181 |
|
| 182 |
+
# Data Loading
|
| 183 |
eligibility_df = pd.read_csv("cards_eligibility_updated.csv")
|
| 184 |
df = pd.read_csv("credit_card_data_updated.csv")
|
| 185 |
# print(df.head())
|
|
|
|
| 190 |
all_card_lookup = dict(zip(df_all_cards["name"], df_all_cards["description"]))
|
| 191 |
|
| 192 |
|
| 193 |
+
# Eligibility Lookup
|
| 194 |
eligibility_lookup = {}
|
| 195 |
for _, row in eligibility_df.iterrows():
|
| 196 |
name = row["Name"].strip()
|