Update app.py

app.py

@@ -1,20 +1,20 @@
 import gradio as gr
 
 import os
-import google.generativeai as palm
+import google.generativeai as genai
 
-palm.configure(
+genai.configure(
     api_key=os.getenv("api_key")
 )
 
-model_bison = "models/chat-bison-001"
+models = "gemini-pro"
 
-def generate_text(prompt, examples, model=model_bison, temperature=0.25):
-    return palm.chat(model=model, temperature=temperature, examples=examples, messages=prompt).last
+def generate_text(prompt, examples, model=models, temperature=0.25):
+    return genai.chat(model=model, temperature=temperature, examples=examples, messages=prompt).last
 
 def generate(prompt, history, temperature=0.25):
     examples = [(item[0], item[1]) for item in history]
-    output = generate_text(prompt, examples, model_bison, temperature)
+    output = generate_text(prompt, examples, models, temperature)
     return output
 
 additional_inputs=[
@@ -38,7 +38,7 @@ css = """
 """
 
 with gr.Blocks(css=css) as demo:
-    gr.HTML("<h1><center>PaLM 2 Chat by <a href='https://rishiraj.github.io/'>Rishiraj Acharya</a></center></h1>")
+    gr.HTML("<h1><center>LLaVa Chat by a href='https://huggingface.co/Satyam-Singh'>Satyam Singh</a></center></h1>")
     gr.ChatInterface(
         generate,
         additional_inputs=additional_inputs,