Update main.py

main.py

@@ -10,8 +10,7 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from typing import List, Dict, Any, Optional
 from datetime import datetime
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import StreamingResponse  # Add this import
+from fastapi.responses import StreamingResponse
 
 # Mock implementations for ImageResponse and to_data_uri
 class ImageResponse:
@@ -171,7 +170,7 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                 else:
                     async for chunk in response.content.iter_any():
                         if chunk:
-                            decoded_chunk = chunk.decode()
+                            decoded_chunk = chunk.decode(errors='ignore')  # Handle decoding errors
                             decoded_chunk = re.sub(r'\$@\$v=[^$]+\$@\$', '', decoded_chunk)
                             if decoded_chunk.strip():
                                 yield decoded_chunk