bibibi12345 commited on
Commit
138e428
·
1 Parent(s): e20783b

fixing streaming and safetysettings

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -93,7 +93,7 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
93
  request_json["generationConfig"]
94
  request_json["generationConfig"]["responseModalities"] = ["TEXT", "IMAGE"]
95
 
96
- request_json["generationConfig"]["safetySettings"] = [
97
  {
98
  "category": "HARM_CATEGORY_HATE_SPEECH",
99
  "threshold": "OFF"
 
93
  request_json["generationConfig"]
94
  request_json["generationConfig"]["responseModalities"] = ["TEXT", "IMAGE"]
95
 
96
+ request_json["safetySettings"] = [
97
  {
98
  "category": "HARM_CATEGORY_HATE_SPEECH",
99
  "threshold": "OFF"