niladridutta commited on
Commit
8e7e236
1 Parent(s): c4bc583

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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
- generation_config = config1,
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