jjjulllesss commited on
Commit
7a0516a
·
verified ·
1 Parent(s): 8d1968e

env variable & model update

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -74,7 +74,7 @@ class AudioToTextTool(Tool):
74
  # Set up the API URL and headers
75
  api_url = "https://router.huggingface.co/hf-inference/models/openai/whisper-large-v3-turbo"
76
  headers = {
77
- "Authorization": f"Bearer {os.getenv('HF_API_KEY')}",
78
  "Content-Type": "audio/mpeg" # Assuming MP3 format
79
  }
80
 
@@ -104,7 +104,7 @@ class ImageAnalysisTool(Tool):
104
  )
105
 
106
  completion = client.chat.completions.create(
107
- model="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
108
  messages=[
109
  {
110
  "role": "user",
 
74
  # Set up the API URL and headers
75
  api_url = "https://router.huggingface.co/hf-inference/models/openai/whisper-large-v3-turbo"
76
  headers = {
77
+ "Authorization": f"Bearer {os.getenv('HF_TOKEN')}",
78
  "Content-Type": "audio/mpeg" # Assuming MP3 format
79
  }
80
 
 
104
  )
105
 
106
  completion = client.chat.completions.create(
107
+ model="google/gemma-3-27b-it",
108
  messages=[
109
  {
110
  "role": "user",