Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,8 @@ load_dotenv()
|
|
13 |
# Configuration
|
14 |
MODEL_NAME = "all-MiniLM-L6-v2"
|
15 |
GENAI_MODEL = "gemini-pro"
|
16 |
-
DATASET_LINK = "
|
|
|
17 |
CHUNK_SIZE = 500
|
18 |
TOP_K = 3
|
19 |
|
@@ -28,7 +29,7 @@ class GeminiRAGSystem:
|
|
28 |
self.index = None
|
29 |
self.chunks = []
|
30 |
self.dataset_loaded = False
|
31 |
-
self.gemini_api_key = os.getenv("
|
32 |
if self.gemini_api_key:
|
33 |
genai.configure(api_key=self.gemini_api_key)
|
34 |
|
|
|
13 |
# Configuration
|
14 |
MODEL_NAME = "all-MiniLM-L6-v2"
|
15 |
GENAI_MODEL = "gemini-pro"
|
16 |
+
DATASET_LINK = "curl -X GET \
|
17 |
+
"https://datasets-server.huggingface.co/rows?dataset=midrees2806%2F7K_Dataset&config=default&split=train&offset=100&length=100""
|
18 |
CHUNK_SIZE = 500
|
19 |
TOP_K = 3
|
20 |
|
|
|
29 |
self.index = None
|
30 |
self.chunks = []
|
31 |
self.dataset_loaded = False
|
32 |
+
self.gemini_api_key = os.getenv("AIzaSyASrFvE3gFPigihza0JTuALzZmBx0Kc3d0")
|
33 |
if self.gemini_api_key:
|
34 |
genai.configure(api_key=self.gemini_api_key)
|
35 |
|