Update app.py

app.py

@@ -54,7 +54,7 @@ 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)
+    gen_model = client.models.get(model=model)
     
     # Prepare inputs
     if model_types[model] == "text" and files:
@@ -107,7 +107,7 @@ 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)
+    gen_model = client.models.get(model=model)
     
     # Prepare inputs
     if model_types[model] == "text" and files: