Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 
 anyscale_base_url = "https://api.endpoints.anyscale.com/v1"
 
-def chat_with_openai(api_key, user_input):
+def predict(api_key, user_input):
     prompt = f"""[INST] {user_input} [/INST]"""
     
     client = OpenAI(base_url=anyscale_base_url, api_key=api_key)
@@ -20,12 +20,12 @@ def main():
     description = "This is a simple interface to interact with OpenAI’s Chat Completion API. Please enter your API key and your message."
     with gr.Blocks() as demo:
         with gr.Row():
-            api_key_input = gr.Textbox(label="API Key", placeholder="Enter your OpenAI API key here", show_label=True, type="password")
+            api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here", show_label=True, type="password")
             user_input = gr.Textbox(label="Your Message", placeholder="Enter your message here")
             submit_btn = gr.Button("Submit")
         output = gr.Textbox(label="Chatbot Response")
         
-        submit_btn.click(fn=chat_with_openai, inputs=[api_key_input, user_input], outputs=output)
+        submit_btn.click(fn=predict, inputs=[api_key_input, user_input], outputs=output)
 
     demo.launch()