gmustafa413 commited on
Commit
86f28ce
·
verified ·
1 Parent(s): 3c411f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = "https://huggingface.co/datasets/midrees2806/7K_Dataset"
 
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("GEMINI_API_KEY")
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