Spaces:

shimer56
/

Extract_PDF

Sleeping

shimer56 commited on Jun 22

Commit

04e96c0

•

1 Parent(s): 2591f57

Upload folder using huggingface_hub

Files changed (1) hide show

extract_images/services.py CHANGED Viewed

@@ -95,7 +95,8 @@ def extract_images_gemini(model, pdf_file):
 def extract_images_gpt(model, pdf_file):
-    openai.api_key = OPENAI_API_KEY
     image_media_type = "image/png"
     pdf_path = "extract_images/input_docs/uploaded_pdf.pdf"

 def extract_images_gpt(model, pdf_file):
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    openai.api_key = openai_api_key
     image_media_type = "image/png"
     pdf_path = "extract_images/input_docs/uploaded_pdf.pdf"