Spaces:
Running
Running
Commit
·
ca2603f
1
Parent(s):
2b9f3af
fix: change to load env from local.
Browse files
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.
|
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.
|
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={
|