Update app.py

app.py

@@ -13,7 +13,7 @@ import google.generativeai as gemini
 gemini.configure(api_key='AIzaSyArybMiPDZARDCz7yZBjzEMx6zXgOXHtoc')
 # Use the palm.list_models function to find available models
 # PaLM 2 available in 4 sizes: Gecko, Otter, Bison and Unicorn (largest)
-models = [m for m in gemini.list_models() if 'generateText' in m.supported_generation_methods]
+models =gemini.GenerativeModel(gemini-1.0-pro)
 model = models[0].name
 # define completion function
 def get_completion(code_snippet):