danielkorat commited on
Commit
5b2afc5
·
verified ·
1 Parent(s): 9360834

Update scripts/visual_qa.py

Browse files
Files changed (1) hide show
  1. scripts/visual_qa.py +1 -1
scripts/visual_qa.py CHANGED
@@ -119,7 +119,7 @@ class VisualQATool(Tool):
119
  }
120
  output_type = "string"
121
 
122
- client = InferenceClient("HuggingFaceM4/idefics2-8b-chatty")
123
 
124
  def forward(self, image_path: str, question: Optional[str] = None) -> str:
125
  output = ""
 
119
  }
120
  output_type = "string"
121
 
122
+ client = InferenceClient("HuggingFaceM4/idefics2-8b-chatty", token=os.environ["HUB_TOKEN"])
123
 
124
  def forward(self, image_path: str, question: Optional[str] = None) -> str:
125
  output = ""