opentyphoon commited on
Commit
ca2603f
·
1 Parent(s): 2b9f3af

fix: change to load env from local.

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -4,14 +4,12 @@ import json
4
  import os
5
  from meta_prompt import get_prompt
6
  from openai import OpenAI
7
- from dotenv import load_dotenv
8
  from utils import render_pdf_to_base64png, image_to_pdf, get_anchor_text
9
  import gradio as gr
10
  from PIL import Image
11
 
12
- load_dotenv()
13
 
14
- openai = OpenAI(base_url=os.getenv("TYPHOON_BASE_URL"), api_key=os.getenv("TYPHOON_API_KEY"))
15
 
16
  theme = gr.themes.Soft(
17
  primary_hue=gr.themes.Color(
@@ -66,7 +64,7 @@ def process_pdf(pdf_or_image_file, task_type):
66
  }]
67
  # send messages to openai compatible api
68
  response = openai.chat.completions.create(
69
- model=os.getenv("TYPHOON_OCR_MODEL"),
70
  messages=messages,
71
  max_tokens=16384,
72
  extra_body={
 
4
  import os
5
  from meta_prompt import get_prompt
6
  from openai import OpenAI
 
7
  from utils import render_pdf_to_base64png, image_to_pdf, get_anchor_text
8
  import gradio as gr
9
  from PIL import Image
10
 
 
11
 
12
+ openai = OpenAI(base_url=os.environ.get("TYPHOON_BASE_URL"), api_key=os.environ.get("TYPHOON_API_KEY"))
13
 
14
  theme = gr.themes.Soft(
15
  primary_hue=gr.themes.Color(
 
64
  }]
65
  # send messages to openai compatible api
66
  response = openai.chat.completions.create(
67
+ model=os.environ.get("TYPHOON_OCR_MODEL"),
68
  messages=messages,
69
  max_tokens=16384,
70
  extra_body={