arjunanand13 commited on
Commit
19e3d1d
·
verified ·
1 Parent(s): aa82c6e

Update qa_bot_chatgpt.py

Browse files
Files changed (1) hide show
  1. qa_bot_chatgpt.py +3 -0
qa_bot_chatgpt.py CHANGED
@@ -15,6 +15,9 @@ class QAInfer:
15
  # self.chatgpt_client = OpenAI(api_key="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")
16
 
17
  self.chatgpt_client = OpenAI(api_key="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")
 
 
 
18
  def extract_text_from_pdf(self, pdf_path):
19
  """Extract text from a PDF file."""
20
  reader = PdfReader(pdf_path)
 
15
  # self.chatgpt_client = OpenAI(api_key="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")
16
 
17
  self.chatgpt_client = OpenAI(api_key="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")
18
+ self.genai.configure(api_key="AIzaSyAFG94rVbm9eWepO5uPGsMha8XJ-sHbMdA")
19
+ self.genai_model = genai.GenerativeModel('gemini-pro')
20
+
21
  def extract_text_from_pdf(self, pdf_path):
22
  """Extract text from a PDF file."""
23
  reader = PdfReader(pdf_path)