Update app.py
Browse files
app.py
CHANGED
|
@@ -36,6 +36,7 @@ safety_settings = [
|
|
| 36 |
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
| 37 |
}
|
| 38 |
]
|
|
|
|
| 39 |
@st.cache_resource
|
| 40 |
def load_model(model1,config1,safety1):
|
| 41 |
return genai.GenerativeModel(model_name = model1, generation_config = config1, safety_settings = safety1)
|
|
|
|
| 36 |
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
| 37 |
}
|
| 38 |
]
|
| 39 |
+
|
| 40 |
@st.cache_resource
|
| 41 |
def load_model(model1,config1,safety1):
|
| 42 |
return genai.GenerativeModel(model_name = model1, generation_config = config1, safety_settings = safety1)
|