Update app.py

app.py

@@ -1,16 +1,12 @@
 # -*- coding: utf-8 -*-
 
-# Import necessary libraries
 import gradio as gr
 import google.generativeai as genai
 
 # Configure the API key
 genai.configure(api_key='AIzaSyYourGenAIKeyHere')
 
-# Load the generative model using the GenerativeModel class
-model = genai.GenerativeModel(name="gemini", version="v1")
-
-# Define the completion function using generate_content
+# Define the completion function using the unified generate_content method
 def get_completion(code_snippet):
     python_code_examples = """
     ---------------------
@@ -56,7 +52,7 @@ def get_completion(code_snippet):
     """
 
     # Generate content using the unified method
-    completion = model.generate_content(
+    completion = genai.generate_content(
         prompt=prompt,
         temperature=0,
         max_output_tokens=500,