Update app.py
Browse files
app.py
CHANGED
@@ -132,7 +132,7 @@ def bot(
|
|
132 |
response = model.generate_content(
|
133 |
prompt,
|
134 |
stream=True,
|
135 |
-
generation_config=generation_config,
|
136 |
response.resolve()
|
137 |
|
138 |
# out1 = response.text
|
@@ -142,7 +142,7 @@ def bot(
|
|
142 |
# response = model.generate_content(
|
143 |
# prompt,
|
144 |
# stream=True,
|
145 |
-
# generation_config=generation_config,
|
146 |
response.resolve()
|
147 |
else:
|
148 |
image_prompt = preprocess_image(image_prompt)
|
@@ -150,7 +150,7 @@ def bot(
|
|
150 |
response = model.generate_content(
|
151 |
contents=[text_prompt, image_prompt],
|
152 |
stream=True,
|
153 |
-
generation_config=generation_config,
|
154 |
response.resolve()
|
155 |
|
156 |
# streaming effect
|
|
|
132 |
response = model.generate_content(
|
133 |
prompt,
|
134 |
stream=True,
|
135 |
+
generation_config=generation_config,safety_settings=safety_settings)
|
136 |
response.resolve()
|
137 |
|
138 |
# out1 = response.text
|
|
|
142 |
# response = model.generate_content(
|
143 |
# prompt,
|
144 |
# stream=True,
|
145 |
+
# generation_config=generation_config, safety_settings=afety_settings)
|
146 |
response.resolve()
|
147 |
else:
|
148 |
image_prompt = preprocess_image(image_prompt)
|
|
|
150 |
response = model.generate_content(
|
151 |
contents=[text_prompt, image_prompt],
|
152 |
stream=True,
|
153 |
+
generation_config=generation_config, safety_settings=safety_settings)
|
154 |
response.resolve()
|
155 |
|
156 |
# streaming effect
|