Spaces:
Runtime error
Runtime error
Commit
•
ea8bd6e
1
Parent(s):
27e552a
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,9 +11,7 @@ generation_config = {
|
|
| 11 |
"top_k": 1,
|
| 12 |
"max_output_tokens": 4096,
|
| 13 |
}
|
| 14 |
-
|
| 15 |
-
generation_config=generation_config,
|
| 16 |
-
safety_settings=safety_settings)
|
| 17 |
safety_settings = [
|
| 18 |
{
|
| 19 |
"category": "HARM_CATEGORY_HARASSMENT",
|
|
@@ -33,6 +31,9 @@ safety_settings = [
|
|
| 33 |
},
|
| 34 |
]
|
| 35 |
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
convo = model.start_chat(history=[
|
| 38 |
{
|
|
|
|
| 11 |
"top_k": 1,
|
| 12 |
"max_output_tokens": 4096,
|
| 13 |
}
|
| 14 |
+
|
|
|
|
|
|
|
| 15 |
safety_settings = [
|
| 16 |
{
|
| 17 |
"category": "HARM_CATEGORY_HARASSMENT",
|
|
|
|
| 31 |
},
|
| 32 |
]
|
| 33 |
|
| 34 |
+
model = genai.GenerativeModel(model_name="gemini-pro",
|
| 35 |
+
generation_config=generation_config,
|
| 36 |
+
safety_settings=safety_settings)
|
| 37 |
|
| 38 |
convo = model.start_chat(history=[
|
| 39 |
{
|