shubham7080 commited on
Commit
6cb4b8e
·
verified ·
1 Parent(s): d59cd2a

Update tools/describe_image_tool.py

Browse files
Files changed (1) hide show
  1. tools/describe_image_tool.py +9 -1
tools/describe_image_tool.py CHANGED
@@ -4,7 +4,15 @@ import os
4
  from openai import OpenAI
5
  from smolagents import Tool
6
 
7
- client = OpenAI()
 
 
 
 
 
 
 
 
8
 
9
 
10
  class DescribeImageTool(Tool):
 
4
  from openai import OpenAI
5
  from smolagents import Tool
6
 
7
+ import os
8
+ from smolagents import AzureOpenAIServerModel
9
+
10
+ model = AzureOpenAIServerModel(
11
+ model_id = os.environ.get("AZURE_OPENAI_MODEL"),
12
+ azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT"),
13
+ api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
14
+ api_version=os.environ.get("OPENAI_API_VERSION")
15
+ )
16
 
17
 
18
  class DescribeImageTool(Tool):