Update app.py

app.py

@@ -54,7 +54,6 @@ def process_files(files, model_type):
 def chat_submit_func(message, files, chat_history, model, temperature, top_p, max_tokens, api_key):
     print(model)
     client = genai.Client(api_key=api_key)
-    gen_model = client.models.get(model=model)
     
     # Prepare inputs
     if model_types[model] == "text" and files:
@@ -73,7 +72,7 @@ def chat_submit_func(message, files, chat_history, model, temperature, top_p, ma
     }
     
     try:
-        response = gen_model.generate_content(inputs, generation_config=generation_config)
+        response = client.models.generate_content(inputs, model=model, generation_config=generation_config)
         response_text = ""
         response_images = []
         
@@ -107,7 +106,6 @@ def chat_submit_func(message, files, chat_history, model, temperature, top_p, ma
 def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api_key):
     print(model)
     client = genai.Client(api_key=api_key)
-    gen_model = client.models.get(model=model)
     
     # Prepare inputs
     if model_types[model] == "text" and files:
@@ -127,7 +125,7 @@ def single_submit_func(prompt, files, model, temperature, top_p, max_tokens, api
     }
     
     try:
-        response = gen_model.generate_content(inputs, generation_config=generation_config)
+        response = client.models.generate_content(inputs, model=model, generation_config=generation_config)
         response_text = warning
         response_images = []