Update app.py

app.py

@@ -1,15 +1,31 @@
+import os
 import gradio as gr
-from transformers import pipeline
+import openai
+
+# Set up OpenAI API
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def chat_with_gpt(input_text):
-    model_name = "gpt2-large" # Replace this with the appropriate GPT model you want to use.
-    chat_gpt = pipeline("text-generation", model=model_name, device=0)
-    generated_text = chat_gpt(input_text, max_length=150, do_sample=True, top_k=50, top_p=0.95)
-    return generated_text[0]["generated_text"]
+    model_engine = "text-davinci-002" # Choose the appropriate model engine.
+    prompt = f"{input_text} (AI):"
+
+    response = openai.Completion.create(
+        engine=model_engine,
+        prompt=prompt,
+        max_tokens=100,
+        n=1,
+        stop=None,
+        temperature=0.8,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
+
+    generated_text = response.choices[0].text.strip()
+    return generated_text
 
 # Gradio user interface
 input_text = gr.inputs.Textbox(lines=5, label="Your question about AI:")
 output_text = gr.outputs.Textbox(label="ChatGPT Response:")
 
-iface = gr.Interface(fn=chat_with_gpt, inputs=input_text, outputs=output_text, title="Chat with ChatGPT", description="Ask ChatGPT questions about AI.")
-iface.launch()
+iface = gr.Interface(fn=chat_with_gpt)