niladridutta
commited on
Commit
•
8e7e236
1
Parent(s):
c4bc583
Update app.py
Browse files
app.py
CHANGED
@@ -38,9 +38,8 @@ safety_settings = [
|
|
38 |
]
|
39 |
@st.cache_resource
|
40 |
def load_model(model1,config1,safety1):
|
41 |
-
return genai.GenerativeModel(model_name = model1,
|
42 |
-
|
43 |
-
safety_settings = safety1)
|
44 |
model = load_model("gemini-pro",generation_config,safety_settings)
|
45 |
|
46 |
|
|
|
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)
|
42 |
+
|
|
|
43 |
model = load_model("gemini-pro",generation_config,safety_settings)
|
44 |
|
45 |
|