thanhnt-cf commited on
Commit
923ae0c
·
1 Parent(s): 7a38b33

do not use dotenv on huggingface spaces

Browse files
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import json
2
  import gradio as gr
3
- from dotenv import load_dotenv
4
  from llm_api.exceptions import RefusalError
5
  # from llm_api.langchain_api import extract_info, follow_structure
6
  from llm_api.native_api import extract_info, follow_structure
7
  from openai import BadRequestError
8
 
9
- load_dotenv('.env', override=True)
10
 
11
  def extract_free_fn(gallery, provider, schema):
12
  img_paths = [item[0] for item in gallery]
 
1
  import json
2
  import gradio as gr
3
+ # from dotenv import load_dotenv
4
  from llm_api.exceptions import RefusalError
5
  # from llm_api.langchain_api import extract_info, follow_structure
6
  from llm_api.native_api import extract_info, follow_structure
7
  from openai import BadRequestError
8
 
9
+ # load_dotenv('.env', override=True)
10
 
11
  def extract_free_fn(gallery, provider, schema):
12
  img_paths = [item[0] for item in gallery]
llm_api/anthropic_api.py CHANGED
@@ -1,9 +1,9 @@
1
  from anthropic import Anthropic
2
- from dotenv import load_dotenv
3
  from llm_api.utils import get_data_format, get_image_data
4
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
5
 
6
- load_dotenv(override=True)
7
  client = Anthropic()
8
 
9
  def extract_info(img_paths, schema):
 
1
  from anthropic import Anthropic
2
+ # from dotenv import load_dotenv
3
  from llm_api.utils import get_data_format, get_image_data
4
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
5
 
6
+ # load_dotenv(override=True)
7
  client = Anthropic()
8
 
9
  def extract_info(img_paths, schema):
llm_api/langchain_api.py CHANGED
@@ -1,13 +1,13 @@
1
  from .utils import get_data_format, get_image_data
2
  from langchain_anthropic import ChatAnthropic
3
  from langchain_openai import ChatOpenAI
4
- from dotenv import load_dotenv
5
  from langchain_core.messages import HumanMessage, SystemMessage
6
  from schema_structured import Garment as StructuredGarment
7
  from schema_free import Garment as FreeGarment
8
 
9
 
10
- load_dotenv()
11
 
12
 
13
  anthropic_llm = ChatAnthropic(
 
1
  from .utils import get_data_format, get_image_data
2
  from langchain_anthropic import ChatAnthropic
3
  from langchain_openai import ChatOpenAI
4
+ # from dotenv import load_dotenv
5
  from langchain_core.messages import HumanMessage, SystemMessage
6
  from schema_structured import Garment as StructuredGarment
7
  from schema_free import Garment as FreeGarment
8
 
9
 
10
+ # load_dotenv()
11
 
12
 
13
  anthropic_llm = ChatAnthropic(
llm_api/native_api.py CHANGED
@@ -1,5 +1,5 @@
1
  import base64
2
- from dotenv import load_dotenv
3
  from . import openai_api as openai_info_extractor
4
  from . import anthropic_api as anthropic_extractor
5
 
 
1
  import base64
2
+ # from dotenv import load_dotenv
3
  from . import openai_api as openai_info_extractor
4
  from . import anthropic_api as anthropic_extractor
5
 
llm_api/openai_api.py CHANGED
@@ -1,11 +1,11 @@
1
  import json
2
- from dotenv import load_dotenv
3
  from .utils import get_data_format, get_image_data
4
  from openai import OpenAI, BadRequestError
5
  from .exceptions import RefusalError
6
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
7
 
8
- load_dotenv(override=True)
9
  client = OpenAI()
10
 
11
 
 
1
  import json
2
+ # from dotenv import load_dotenv
3
  from .utils import get_data_format, get_image_data
4
  from openai import OpenAI, BadRequestError
5
  from .exceptions import RefusalError
6
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
7
 
8
+ # load_dotenv(override=True)
9
  client = OpenAI()
10
 
11