Update app.py

app.py

@@ -1,21 +1,22 @@
 import gradio as gr
 from transformers import pipeline
 
-# Create a text generation pipeline using Hugging Face Transformers
-generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+# Replace 'your_openai_api_key' with your actual OpenAI API key
+api_key = "your_openai_api_key"
 
-# Define the chat function
-def chat(input_text):
-    response = generator(input_text, max_length=50, do_sample=True)[0]["generated_text"]
-    return response
+def generate_response(text):
+    generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B", device=0)  # You can choose a different model here
+    response = generator(text, max_length=150, num_return_sequences=1, api_key=api_key)
+    return response[0]['generated_text']
 
-# Create a Gradio interface
 iface = gr.Interface(
-    fn=chat,
-    inputs=gr.inputs.Textbox(label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Response"),
+    fn=generate_response,
+    inputs="text",
+    outputs="text",
+    title="GPT-3.5 Turbo Chat",
+    description="An interface to chat with GPT-3.5 Turbo.",
     live=True,
+    theme="huggingface",
 )
 
-# Launch the Gradio interface
 iface.launch()