Update main.py

main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from huggingface_hub import InferenceClient, BadRequestError
+from huggingface_hub import InferenceClient
 
 app = FastAPI()
 
@@ -51,8 +51,6 @@ def generate(item: Item):
         output = "".join([response.token.text for response in stream])
         return output
 
-    except BadRequestError as e:
-        raise HTTPException(status_code=400, detail=f"Bad request: {str(e)}")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Internal error: {str(e)}")