Update app.py

app.py

@@ -30,7 +30,7 @@ def openai_chat(api_key, model, message):
 # Define the chat function for Hugging Face API
 def hf_chat(model_name, message):
     # Load the model and tokenizer
-    model = transformers.pipeline("text2text-generation", model=model_name)
+    model = transformers.pipeline("text-generation", model=model_name)
     
     # Generate a response from the model
     bot_response = model(message, max_length=1024, do_sample=True, temperature=0.7)[0]["generated_text"]
@@ -41,8 +41,8 @@ def hf_chat(model_name, message):
 api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None)
 model_input = gr.inputs.Dropdown(
     label="Select OpenAI model",
-    choices=["text-davinci-003", "text-davinci-002"],
-    default="text-davinci-003",
+    choices=["davinci", "curie", "babbage"],
+    default="davinci",
 )
 message_input = gr.inputs.Textbox(label="Enter your message here")
 output = gr.outputs.Textbox(label="Bot response")