Lasdw commited on
Commit
2e1d581
·
1 Parent(s): 88b0133

Added Image detection via url

Browse files
Files changed (1) hide show
  1. agent.py +3 -0
agent.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  from typing import TypedDict, Annotated, Dict, Any, Optional, Union, List
3
  from langgraph.graph.message import add_messages
4
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage, ToolMessage
@@ -29,6 +30,8 @@ from transformers import AutoProcessor, AutoModelForVision2Seq
29
 
30
  from apify_client import ApifyClient
31
 
 
 
32
  # Initialize vision model and processor globally for reuse
33
  # This avoids reloading the model on each call
34
  try:
 
1
  import os
2
+ from dotenv import load_dotenv
3
  from typing import TypedDict, Annotated, Dict, Any, Optional, Union, List
4
  from langgraph.graph.message import add_messages
5
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage, ToolMessage
 
30
 
31
  from apify_client import ApifyClient
32
 
33
+ load_dotenv()
34
+
35
  # Initialize vision model and processor globally for reuse
36
  # This avoids reloading the model on each call
37
  try: