Spaces:
Sleeping
Sleeping
Update scripts/visual_qa.py
Browse files- 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 = ""
|